edadaltocg commited on
Commit
e87c581
1 Parent(s): 1b4bd62

Upload hyperparameters.json

Browse files
Files changed (1) hide show
  1. hyperparameters.json +28 -0
hyperparameters.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "config": "scripts/train_configs/svhn.json",
3
+ "model": "resnet18_svhn",
4
+ "dataset": "svhn",
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.01,
13
+ "optimizer_kwargs": {
14
+ "momentum": 0.9,
15
+ "weight_decay": 0.0005
16
+ },
17
+ "scheduler": "MultiStepLR",
18
+ "scheduler_kwargs": {
19
+ "gamma": 0.1,
20
+ "milestones": [
21
+ 75,
22
+ 100,
23
+ 150,
24
+ 225
25
+ ]
26
+ },
27
+ "debug": false
28
+ }