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