From 6ad8238b5aeebf4ba14286448fd719cdb412fc54 Mon Sep 17 00:00:00 2001 From: Mobinapournemat <45605451+Mobinapournemat@users.noreply.github.com> Date: Fri, 29 Dec 2023 14:25:29 +0100 Subject: [PATCH] Update finetune_hela.py --- examples/finetuning/finetune_hela.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/finetuning/finetune_hela.py b/examples/finetuning/finetune_hela.py index 0b8de7920..58a34b51b 100644 --- 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: