Skip to content

Commit 7f34968

Browse files
committed
Fix patch_size
Signed-off-by: Behrooz <[email protected]>
1 parent 0724dfe commit 7f34968

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pathology/multiple_instance_learning/panda_mil_train_evaluate_pytorch_gpu.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -283,7 +283,7 @@ def main_worker(gpu, args):
283283
LabelEncodeIntegerGraded(keys=["label"], num_classes=args.num_classes),
284284
RandGridPatchd(
285285
keys=["image"],
286-
patch_size=args.tile_size,
286+
patch_size=(args.tile_size, args.tile_size),
287287
max_num_patches=args.tile_count,
288288
sort_key="min",
289289
pad_opts={"constant_values": 255},
@@ -302,7 +302,7 @@ def main_worker(gpu, args):
302302
LabelEncodeIntegerGraded(keys=["label"], num_classes=args.num_classes),
303303
GridPatchd(
304304
keys=["image"],
305-
patch_size=args.tile_size,
305+
patch_size=(args.tile_size, args.tile_size),
306306
sort_key="min",
307307
pad_opts={"constant_values": 255},
308308
),

0 commit comments

Comments
 (0)