edadaltocg commited on
Commit
2b766dd
1 Parent(s): 6880aa3

Upload hyperparameters.json

Browse files
Files changed (1) hide show
  1. hyperparameters.json +26 -0
hyperparameters.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "config": "scripts/train_configs/cifar10.json",
3
+ "model": "vgg16_bn_cifar10",
4
+ "dataset": "cifar10",
5
+ "batch_size": 128,
6
+ "epochs": 300,
7
+ "validation_frequency": 5,
8
+ "seed": 1,
9
+ "criterion": "CrossEntropyLoss",
10
+ "criterion_kwargs": {},
11
+ "optimizer": "SGD",
12
+ "lr": 0.1,
13
+ "optimizer_kwargs": {
14
+ "momentum": 0.9,
15
+ "weight_decay": 0.0005,
16
+ "nesterov": "True"
17
+ },
18
+ "scheduler": "ReduceLROnPlateau",
19
+ "scheduler_kwargs": {
20
+ "factor": 0.1,
21
+ "patience": 3,
22
+ "threshold": 0.001,
23
+ "mode": "max"
24
+ },
25
+ "debug": false
26
+ }