Adding doc for using samplers in finetuning (#571)

computational-cell-analytics · May 1, 2024 · e4886fe · e4886fe
1 parent 11c9a83
commit e4886fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(