Add sort_key

Signed-off-by: Behrooz <[email protected]>

Author: bhashemian

pathology/multiple_instance_learning/panda_mil_train_evaluate_pytorch_gpu.py

@@ -285,6 +285,7 @@ def main_worker(gpu, args):
                 keys=["image"],
                 patch_size=args.tile_size,
                 max_num_patches=args.tile_count,
+                sort_key="max",
                 pad_opts={"constant_values": 255},
             ),
             RandFlipd(keys=["image"], spatial_axis=0, prob=0.5),
@@ -303,6 +304,7 @@ def main_worker(gpu, args):
                 keys=["image"],
                 patch_size=args.tile_size,
                 max_num_patches=args.tile_count,
+                sort_key="max",
                 pad_opts={"constant_values": 255},
             ),
             ScaleIntensityRanged(keys=["image"], a_min=np.float32(255), a_max=np.float32(0)),