monai
medical
katielink commited on
Commit
251565d
1 Parent(s): bbaff61

update params to supprot torch.jit.trace torchscript conversion

Browse files
configs/inference.json CHANGED
@@ -15,7 +15,7 @@
15
  "in_channels": 1,
16
  "out_channels": 14,
17
  "feature_size": 48,
18
- "use_checkpoint": true
19
  },
20
  "network": "$@network_def.to(@device)",
21
  "preprocessing": {
 
15
  "in_channels": 1,
16
  "out_channels": 14,
17
  "feature_size": 48,
18
+ "use_checkpoint": false
19
  },
20
  "network": "$@network_def.to(@device)",
21
  "preprocessing": {
configs/metadata.json CHANGED
@@ -1,7 +1,8 @@
1
  {
2
  "schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_20220324.json",
3
- "version": "0.4.0",
4
  "changelog": {
 
5
  "0.4.0": "add name tag",
6
  "0.3.9": "use ITKreader to avoid mass logs at image loading",
7
  "0.3.8": "restructure readme to match updated template",
 
1
  {
2
  "schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_20220324.json",
3
+ "version": "0.4.1",
4
  "changelog": {
5
+ "0.4.1": "update params to supprot torch.jit.trace torchscript conversion",
6
  "0.4.0": "add name tag",
7
  "0.3.9": "use ITKreader to avoid mass logs at image loading",
8
  "0.3.8": "restructure readme to match updated template",
configs/train.json CHANGED
@@ -19,7 +19,7 @@
19
  "in_channels": 1,
20
  "out_channels": 14,
21
  "feature_size": 48,
22
- "use_checkpoint": true
23
  },
24
  "network": "$@network_def.to(@device)",
25
  "loss": {
 
19
  "in_channels": 1,
20
  "out_channels": 14,
21
  "feature_size": 48,
22
+ "use_checkpoint": false
23
  },
24
  "network": "$@network_def.to(@device)",
25
  "loss": {