update params to supprot torch.jit.trace torchscript conversion
Browse files- configs/inference.json +1 -1
- configs/metadata.json +2 -1
- configs/train.json +1 -1
configs/inference.json
CHANGED
@@ -15,7 +15,7 @@
|
|
15 |
"in_channels": 1,
|
16 |
"out_channels": 14,
|
17 |
"feature_size": 48,
|
18 |
-
"use_checkpoint":
|
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.
|
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":
|
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": {
|