Rename em generalist training script

computational-cell-analytics · Oct 25, 2023 · 4e86a21 · 4e86a21
1 parent 32dae45
commit 4e86a21
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/finetuning/generalists/training/electron_microscopy/obtain_em_datasets.py b/finetuning/generalists/training/electron_microscopy/obtain_em_datasets.py
@@ -76,12 +76,11 @@ def _check_dataset_available_for_rois(path, patch_shape):
     datasets.get_platynereis_cell_dataset(path=os.path.join(path, "platynereis"), patch_shape=patch_shape, download=True)
     datasets.get_platynereis_nuclei_dataset(path=os.path.join(path, "platynereis"), patch_shape=patch_shape, download=True)
     datasets.get_platynereis_cilia_dataset(path=os.path.join(path, "platynereis"), patch_shape=patch_shape, download=True)
-    print("All the datasets are available")
+    print("All the datasets are available for RoI splitting")
 
 
 def get_concat_em_datasets(input_path, patch_shape):
     _check_dataset_available_for_rois(path=input_path, patch_shape=patch_shape)
-    breakpoint()
 
     sampler = MinInstanceSampler()
 

diff --git a/...ectron_microscopy/train_em_generalists.py → ...lectron_microscopy/train_em_generalist.py b/...ectron_microscopy/train_em_generalists.py → ...lectron_microscopy/train_em_generalist.py