update scripts (#825)

Signed-off-by: Dong Yang <[email protected]>

Co-authored-by: Dong Yang <[email protected]>

Author: dongyang0122

automl/DiNTS/arch_code.pth

@@ -280,7 +280,7 @@ def main():
             CopyItemsd(keys=["label"], times=1, names=["label4crop"]),
             Lambdad(
                 keys=["label4crop"],
-                func=lambda x: np.concatenate(tuple([ndimage.binary_dilation((x==_k).astype(x.dtype), iterations=48).astype(x.dtype) for _k in range(output_classes)]), axis=0),
+                func=lambda x: np.concatenate(tuple([ndimage.binary_dilation((x==_k).astype(x.dtype), iterations=48).astype(float) for _k in range(output_classes)]), axis=0),
                 overwrite=True,
             ),
             EnsureTyped(keys=["image", "label"]),

automl/DiNTS/search_dints.py

@@ -9,7 +9,7 @@ JSON_PATH="${DATA_ROOT}/dataset.json"
 FOLD=4
 NUM_FOLDS=5
 
-NUM_GPUS_PER_NODE=4
+NUM_GPUS_PER_NODE=8
 NUM_NODES=1
 
 if [ ${NUM_GPUS_PER_NODE} -eq 1 ]

automl/DiNTS/search_dints.sh

@@ -268,7 +268,7 @@ def main():
             CopyItemsd(keys=["label"], times=1, names=["label4crop"]),
             Lambdad(
                 keys=["label4crop"],
-                func=lambda x: np.concatenate(tuple([ndimage.binary_dilation((x==_k).astype(x.dtype), iterations=48).astype(x.dtype) for _k in range(output_classes)]), axis=0),
+                func=lambda x: np.concatenate(tuple([ndimage.binary_dilation((x==_k).astype(x.dtype), iterations=48).astype(float) for _k in range(output_classes)]), axis=0),
                 overwrite=True,
             ),
             EnsureTyped(keys=["image", "label"]),

automl/DiNTS/train_dints.py

@@ -3,7 +3,7 @@ clear
 
 TASK="Task09_Spleen"
 
-ARCH_CKPT="arch_code.pth"
+ARCH_CKPT="arch_code_cvpr.pth"
 DATA_ROOT="/workspace/data_msd/${TASK}"
 JSON_PATH="${DATA_ROOT}/dataset.json"