File tree Expand file tree Collapse file tree 2 files changed +5
-4
lines changed
modules/bundles/spleen_segmentation Expand file tree Collapse file tree 2 files changed +5
-4
lines changed Original file line number Diff line number Diff line change 7
7
"determinism" : " $monai.utils.set_determinism(seed=123)" ,
8
8
"cudnn_opt" : " $setattr(torch.backends.cudnn, 'benchmark', True)" ,
9
9
"device" : " $torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')" ,
10
- "ckpt_path " : " /workspace/data/tutorials/modules/bundles/spleen_segmentation/models/model.pt " ,
10
+ "ckpt_dir " : " /workspace/data/tutorials/modules/bundles/spleen_segmentation/models" ,
11
11
"dataset_dir" : " /workspace/data/Task09_Spleen" ,
12
12
"images" : " $list(sorted(glob.glob(@dataset_dir + '/imagesTr/*.nii.gz')))" ,
13
13
"labels" : " $list(sorted(glob.glob(@dataset_dir + '/labelsTr/*.nii.gz')))" ,
201
201
},
202
202
{
203
203
"_target_" : " CheckpointSaver" ,
204
- "save_dir" : " @ckpt_path " ,
204
+ "save_dir" : " @ckpt_dir " ,
205
205
"save_dict" : {"model" : " @network" },
206
- "save_key_metric" : true
206
+ "save_key_metric" : true ,
207
+ "key_metric_filename" : " model.pt"
207
208
}
208
209
],
209
210
"key_metric" : {
Original file line number Diff line number Diff line change @@ -50,7 +50,7 @@ python -m monai.bundle verify_net_in_out network_def --meta_file configs/metadat
50
50
Export checkpoint to TorchScript file:
51
51
52
52
```
53
- python -m monai.bundle export network_def --filepath models/model.ts --ckpt_file models/model.pt --meta_file configs/metadata.json --config_file configs/inference.json
53
+ python -m monai.bundle ckpt_export network_def --filepath models/model.ts --ckpt_file models/model.pt --meta_file configs/metadata.json --config_file configs/inference.json
54
54
```
55
55
56
56
# Disclaimer
You can’t perform that action at this time.
0 commit comments