--- a +++ b/configs/inference/config.json @@ -0,0 +1,15 @@ +{ + "exp_name":"EcgResNet34", + "type": "Runner1D", + "device": "cuda:0", + "model": "EcgResNet34", + "model_path": "./experiments/EcgResNet34/checkpoints/00000635.pth", + + "json": "data/val.json", + "mapping_json": "data/class-mapper.json", + "num_classes": 8, + + "batch_size": 128, + "num_workers": 3, + "exp_dir" : "experiments/" +}