File size: 547 Bytes
aa54e66
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
{
  "config": "scripts/train_configs/cifar10.json",
  "model": "resnet18_cifar10",
  "dataset": "cifar10",
  "batch_size": 128,
  "epochs": 300,
  "validation_frequency": 5,
  "seed": 1,
  "criterion": "CrossEntropyLoss",
  "criterion_kwargs": {},
  "optimizer": "SGD",
  "lr": 0.1,
  "optimizer_kwargs": {
    "momentum": 0.9,
    "weight_decay": 0.0005,
    "nesterov": "True"
  },
  "scheduler": "ReduceLROnPlateau",
  "scheduler_kwargs": {
    "factor": 0.1,
    "patience": 3,
    "threshold": 0.001,
    "mode": "max"
  },
  "debug": false
}