Update finetune_hela.py

computational-cell-analytics · Dec 29, 2023 · 6ad8238 · 6ad8238
1 parent 7301a7f
commit 6ad8238
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/finetuning/finetune_hela.py b/examples/finetuning/finetune_hela.py
@@ -46,7 +46,7 @@ def get_dataloader(split, patch_shape, batch_size):
     # raw_key, label_key= None, None
 
     # The 'roi' argument can be used to subselect parts of the data.
-    # Here, we use it to select the first 70 frames fro the test split and the other frames for the val split.
+    # Here, we use it to select the first 70 frames for the train split and the other frames for the val split.
     if split == "train":
         roi = np.s_[:70, :, :]
     else: