From e4886fe50f2e8b68903e2eb5d5fbb0fbc5ff4619 Mon Sep 17 00:00:00 2001
From: Anwai Archit <52396323+anwai98@users.noreply.github.com>
Date: Wed, 1 May 2024 19:28:50 +0200
Subject: [PATCH] Adding doc for using samplers in finetuning (#571)

---
 .../training/light_microscopy/plantseg_root_finetuning.py       | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/finetuning/specialists/training/light_microscopy/plantseg_root_finetuning.py b/finetuning/specialists/training/light_microscopy/plantseg_root_finetuning.py
index 235ab541..723dad79 100644
--- a/finetuning/specialists/training/light_microscopy/plantseg_root_finetuning.py
+++ b/finetuning/specialists/training/light_microscopy/plantseg_root_finetuning.py
@@ -26,7 +26,7 @@ def get_dataloaders(patch_shape, data_path):
     """
     raw_transform = ResizeRawTrafo(patch_shape, do_rescaling=False)
     label_transform = ResizeLabelTrafo(patch_shape)
-    sampler = MinInstanceSampler(min_num_instances=10)
+    sampler = MinInstanceSampler(min_num_instances=10)  # ensures a minimum of 10 instances (inc. background)
     label_dtype = torch.float32
 
     train_loader = get_plantseg_loader(