Skip to content

Commit 788903e

Browse files
committed
Remove strict_check = False in original worker
Seems to be a Colab-specific issue
1 parent 0e69ee4 commit 788903e

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

napari_cellseg3d/code_models/worker_training.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,6 @@ def get_patch_dataset(self, train_transforms):
204204
EnsureChannelFirstd(
205205
keys=["image"],
206206
channel_dim="no_channel",
207-
strict_check=False,
208207
),
209208
RandSpatialCropSamplesd(
210209
keys=["image"],
@@ -242,7 +241,6 @@ def get_dataset_eval(self, eval_dataset_dict):
242241
EnsureChannelFirstd(
243242
keys=["image", "label"],
244243
channel_dim="no_channel",
245-
strict_check=False,
246244
),
247245
# RandSpatialCropSamplesd(
248246
# keys=["image", "label"],
@@ -290,7 +288,6 @@ def get_dataset(self, train_transforms):
290288
EnsureChannelFirstd(
291289
keys=["image"],
292290
channel_dim="no_channel",
293-
strict_check=False,
294291
),
295292
Orientationd(keys=["image"], axcodes="PLI"),
296293
SpatialPadd(

0 commit comments

Comments
 (0)