[fbbdf8]: / configs / training / 1804.06812.json

Download this file

24 lines (20 with data), 394 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
{
"exp_name":"HeartNet2D",
"type": "Trainer2D",
"model": "HeartNet",
"device": "cpu",
"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
}