Nic-Ma commited on
Commit
646c3e1
1 Parent(s): b6adca4

Add imports and checkpoint

Browse files
Files changed (1) hide show
  1. inference.json +11 -5
inference.json CHANGED
@@ -1,9 +1,14 @@
1
  {
 
 
 
 
 
2
  "dataset_dir": "/workspace/data/Task09_Spleen",
3
- "ckpt_path": "/workspace/data//spleen_segmentation/models/model.pt",
4
- "log": "$__import__('logging').basicConfig(level=20)",
5
  "device": "$torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')",
6
- "datalist": "$list(sorted(__import__('glob').glob(@dataset_dir + '/imagesTs/*.nii.gz')))",
7
  "network_def": {
8
  "_target_": "UNet",
9
  "spatial_dims": 3,
@@ -118,18 +123,19 @@
118
  "handlers": [
119
  {
120
  "_target_": "CheckpointLoader",
121
- "_disabled_": "$not __import__('os').path.exists(@ckpt_path)",
 
122
  "load_path": "@ckpt_path",
123
  "load_dict": {"model": "@network"}
124
  },
125
  {
126
  "_target_": "StatsHandler",
127
- "_disabled_": "@log",
128
  "iteration_log": false
129
  }
130
  ],
131
  "evaluator": {
132
  "_target_": "SupervisedEvaluator",
 
133
  "device": "@device",
134
  "val_data_loader": "@dataloader",
135
  "network": "@network",
 
1
  {
2
+ "imports": [
3
+ "$import glob",
4
+ "$import os"
5
+ ],
6
+ "cudnn_opt": "$setattr(torch.backends.cudnn, 'benchmark', True)",
7
  "dataset_dir": "/workspace/data/Task09_Spleen",
8
+ "ckpt_path": "/workspace/data/tutorials/modules/bundles/spleen_segmentation/models/model.pt",
9
+ "download_ckpt": "$monai.apps.utils.download_url('https://huggingface.co/MONAI/example_spleen_segmentation/resolve/main/model.pt', @ckpt_path)",
10
  "device": "$torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')",
11
+ "datalist": "$list(sorted(glob.glob(@dataset_dir + '/imagesTs/*.nii.gz')))",
12
  "network_def": {
13
  "_target_": "UNet",
14
  "spatial_dims": 3,
 
123
  "handlers": [
124
  {
125
  "_target_": "CheckpointLoader",
126
+ "_requires_": "@download_ckpt",
127
+ "_disabled_": "$not os.path.exists(@ckpt_path)",
128
  "load_path": "@ckpt_path",
129
  "load_dict": {"model": "@network"}
130
  },
131
  {
132
  "_target_": "StatsHandler",
 
133
  "iteration_log": false
134
  }
135
  ],
136
  "evaluator": {
137
  "_target_": "SupervisedEvaluator",
138
+ "_requires_": "@cudnn_opt",
139
  "device": "@device",
140
  "val_data_loader": "@dataloader",
141
  "network": "@network",