monai-test
/

swin_unetr_btcv_segmentation

Model card Files Files and versions Community

katielink commited on Aug 16, 2023

Commit

ba7680b

•

1 Parent(s): 251565d

fix train params of use_checkpoint

Files changed (2) hide show

configs/metadata.json +2 -1
configs/train.json +1 -1

configs/metadata.json CHANGED Viewed

@@ -1,7 +1,8 @@
 {
     "schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_20220324.json",
-    "version": "0.4.1",
     "changelog": {
         "0.4.1": "update params to supprot torch.jit.trace torchscript conversion",
         "0.4.0": "add name tag",
         "0.3.9": "use ITKreader to avoid mass logs at image loading",

 {
     "schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_20220324.json",
+    "version": "0.4.2",
     "changelog": {
+        "0.4.2": "fix train params of use_checkpoint",
         "0.4.1": "update params to supprot torch.jit.trace torchscript conversion",
         "0.4.0": "add name tag",
         "0.3.9": "use ITKreader to avoid mass logs at image loading",

configs/train.json CHANGED Viewed

@@ -19,7 +19,7 @@
         "in_channels": 1,
         "out_channels": 14,
         "feature_size": 48,
-        "use_checkpoint": false
     },
     "network": "$@network_def.to(@device)",
     "loss": {

         "in_channels": 1,
         "out_channels": 14,
         "feature_size": 48,
+        "use_checkpoint": true
     },
     "network": "$@network_def.to(@device)",
     "loss": {