Skip to content

Commit 5f284ec

Browse files
committed
Set pad_mode=None
Signed-off-by: Behrooz <[email protected]>
1 parent 975edc4 commit 5f284ec

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

pathology/multiple_instance_learning/panda_mil_train_evaluate_pytorch_gpu.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -284,6 +284,7 @@ def main_worker(gpu, args):
284284
patch_size=(args.tile_size, args.tile_size),
285285
num_patches=args.tile_count,
286286
sort_fn="min",
287+
pad_mode=None,
287288
constant_values=255,
288289
),
289290
RandFlipd(keys=["image"], spatial_axis=0, prob=0.5),
@@ -302,6 +303,7 @@ def main_worker(gpu, args):
302303
keys=["image"],
303304
patch_size=(args.tile_size, args.tile_size),
304305
sort_fn="min",
306+
pad_mode=None,
305307
constant_values=255,
306308
),
307309
ScaleIntensityRanged(keys=["image"], a_min=np.float32(255), a_max=np.float32(0)),

0 commit comments

Comments
 (0)