Skip to content

Commit 583a710

Browse files
committed
[DLMED] update ignite workflows
Signed-off-by: Nic Ma <[email protected]>
1 parent a42ad81 commit 583a710

File tree

2 files changed

+0
-6
lines changed

2 files changed

+0
-6
lines changed

acceleration/distributed_training/unet_evaluation_workflows.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,6 @@
7979
KeepLargestConnectedComponentd,
8080
LoadImaged,
8181
ScaleIntensityd,
82-
EnsureTyped,
8382
SaveImaged,
8483
)
8584

@@ -113,7 +112,6 @@ def evaluate(args):
113112
LoadImaged(keys=["image", "label"]),
114113
AsChannelFirstd(keys=["image", "label"], channel_dim=-1),
115114
ScaleIntensityd(keys="image"),
116-
EnsureTyped(keys=["image", "label"]),
117115
]
118116
)
119117

acceleration/distributed_training/unet_training_workflows.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,6 @@
8181
RandCropByPosNegLabeld,
8282
RandRotate90d,
8383
ScaleIntensityd,
84-
EnsureTyped,
8584
)
8685

8786

@@ -118,7 +117,6 @@ def train(args):
118117
keys=["image", "label"], label_key="label", spatial_size=[96, 96, 96], pos=1, neg=1, num_samples=4
119118
),
120119
RandRotate90d(keys=["image", "label"], prob=0.5, spatial_axes=[0, 2]),
121-
EnsureTyped(keys=["image", "label"]),
122120
]
123121
)
124122

@@ -155,7 +153,6 @@ def train(args):
155153

156154
train_post_transforms = Compose(
157155
[
158-
EnsureTyped(keys="pred"),
159156
Activationsd(keys="pred", sigmoid=True),
160157
AsDiscreted(keys="pred", threshold=0.5),
161158
KeepLargestConnectedComponentd(keys="pred", applied_labels=[1]),
@@ -198,7 +195,6 @@ def main():
198195
train(args=args)
199196

200197

201-
202198
# python -m torch.distributed.launch --nproc_per_node=NUM_GPUS_PER_NODE
203199
# --nnodes=NUM_NODES --node_rank=INDEX_CURRENT_NODE
204200
# --master_addr="192.168.1.1" --master_port=1234

0 commit comments

Comments
 (0)