[DLMED] update smart cache ddp

Nic-Ma · Nic-Ma · commit fa008a832fb9 · 2022-07-11T18:20:43.000+08:00
Signed-off-by: Nic Ma &lt;nma@nvidia.com&gt;
diff --git a/acceleration/distributed_training/unet_training_smartcache.py b/acceleration/distributed_training/unet_training_smartcache.py
@@ -73,7 +73,6 @@
     RandCropByPosNegLabeld,
     RandRotate90d,
     ScaleIntensityd,
-    EnsureTyped,
 )
 
 
@@ -112,7 +111,6 @@ def train(args):
                 keys=["img", "seg"], label_key="seg", spatial_size=[96, 96, 96], pos=1, neg=1, num_samples=4
             ),
             RandRotate90d(keys=["img", "seg"], prob=0.5, spatial_axes=[0, 2]),
-            EnsureTyped(keys=["img", "seg"]),
         ]
     )
 

Original file line number	Diff line number	Diff line change
`@@ -73,7 +73,6 @@`
`73`	`73`	`RandCropByPosNegLabeld,`
`74`	`74`	`RandRotate90d,`
`75`	`75`	`ScaleIntensityd,`
`76`		`- EnsureTyped,`
`77`	`76`	`)`
`78`	`77`
`79`	`78`
`@@ -112,7 +111,6 @@ def train(args):`
`112`	`111`	`keys=["img", "seg"], label_key="seg", spatial_size=[96, 96, 96], pos=1, neg=1, num_samples=4`
`113`	`112`	`),`
`114`	`113`	`RandRotate90d(keys=["img", "seg"], prob=0.5, spatial_axes=[0, 2]),`
`115`		`- EnsureTyped(keys=["img", "seg"]),`
`116`	`114`	`]`
`117`	`115`	`)`
`118`	`116`