Skip to content

Commit 34d4480

Browse files
authored
[DLMED] fix typo (#657)
Signed-off-by: Nic Ma <[email protected]>
1 parent 5b5533c commit 34d4480

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

modules/bundles/spleen_segmentation/configs/train.json

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
"determinism": "$monai.utils.set_determinism(seed=123)",
88
"cudnn_opt": "$setattr(torch.backends.cudnn, 'benchmark', True)",
99
"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",
1111
"dataset_dir": "/workspace/data/Task09_Spleen",
1212
"images": "$list(sorted(glob.glob(@dataset_dir + '/imagesTr/*.nii.gz')))",
1313
"labels": "$list(sorted(glob.glob(@dataset_dir + '/labelsTr/*.nii.gz')))",
@@ -201,9 +201,10 @@
201201
},
202202
{
203203
"_target_": "CheckpointSaver",
204-
"save_dir": "@ckpt_path",
204+
"save_dir": "@ckpt_dir",
205205
"save_dict": {"model": "@network"},
206-
"save_key_metric": true
206+
"save_key_metric": true,
207+
"key_metric_filename": "model.pt"
207208
}
208209
],
209210
"key_metric": {

modules/bundles/spleen_segmentation/docs/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ python -m monai.bundle verify_net_in_out network_def --meta_file configs/metadat
5050
Export checkpoint to TorchScript file:
5151

5252
```
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
5454
```
5555

5656
# Disclaimer

0 commit comments

Comments
 (0)