[c128d9]: / configs / training / 2002.00254.json

Download this file

25 lines (20 with data), 397 Bytes

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