a | b/configs/training/1804.06812.json | ||
---|---|---|---|
1 | { |
||
2 | "exp_name":"HeartNet2D", |
||
3 | "type": "Trainer2D", |
||
4 | "model": "HeartNet", |
||
5 | "device": "cpu", |
||
6 | "train_json": "data/train.json", |
||
7 | "val_json": "data/val.json", |
||
8 | "mapping_json": "data/class-mapper.json", |
||
9 | "num_classes": 8, |
||
10 | |||
11 | "batch_size": 128, |
||
12 | "optim": "Adam", |
||
13 | "optim_params": |
||
14 | { |
||
15 | "lr": 1e-3 |
||
16 | }, |
||
17 | |||
18 | "num_workers": 3, |
||
19 | "exp_dir" : "experiments/", |
||
20 | |||
21 | "model_path": 0, |
||
22 | "epochs": 1 |
||
23 | } |