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