--- a +++ b/configs/training/1707.01836.json @@ -0,0 +1,24 @@ +{ + "exp_name":"HeartNet1D", + "type": "Trainer1D", + "device": "cpu", + "model": "HeartNet", + + "train_json": "data/train.json", + "val_json": "data/val.json", + "mapping_json": "data/class-mapper.json", + "num_classes": 8, + + "batch_size": 128, + "optim": "Adam", + "optim_params": + { + "lr": 1e-3 + }, + + "num_workers": 3, + "exp_dir" : "experiments/", + + "model_path": 0, + "epochs": 1 +}