Skip to content

Commit

Permalink
Merge pull request #294 from Mobinapournemat/patch-1
Browse files Browse the repository at this point in the history
Update finetune_hela.py
constantinpape authored Jan 2, 2024
2 parents 7301a7f + 6ad8238 commit 7651dda
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/finetuning/finetune_hela.py
Original file line number Diff line number Diff line change
@@ -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:

0 comments on commit 7651dda

Please sign in to comment.