Skip to content

Commit 6ad8238

Browse files
Update finetune_hela.py
1 parent 7301a7f commit 6ad8238

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/finetuning/finetune_hela.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def get_dataloader(split, patch_shape, batch_size):
4646
# raw_key, label_key= None, None
4747

4848
# The 'roi' argument can be used to subselect parts of the data.
49-
# Here, we use it to select the first 70 frames fro the test split and the other frames for the val split.
49+
# Here, we use it to select the first 70 frames for the train split and the other frames for the val split.
5050
if split == "train":
5151
roi = np.s_[:70, :, :]
5252
else:

0 commit comments

Comments
 (0)