a | b/configs/inference/config.json | ||
---|---|---|---|
1 | { |
||
2 | "exp_name":"EcgResNet34", |
||
3 | "type": "Runner1D", |
||
4 | "device": "cuda:0", |
||
5 | "model": "EcgResNet34", |
||
6 | "model_path": "./experiments/EcgResNet34/checkpoints/00000635.pth", |
||
7 | |||
8 | "json": "data/val.json", |
||
9 | "mapping_json": "data/class-mapper.json", |
||
10 | "num_classes": 8, |
||
11 | |||
12 | "batch_size": 128, |
||
13 | "num_workers": 3, |
||
14 | "exp_dir" : "experiments/" |
||
15 | } |