edadaltocg commited on
Commit
4005106
1 Parent(s): 6855f4f

Upload hyperparameters.json

Browse files
Files changed (1) hide show
  1. hyperparameters.json +22 -0
hyperparameters.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "config": "scripts/train_configs/ft_cifar10.json",
3
+ "model": "vit_base_patch16_224_in21k_ft_cifar10",
4
+ "dataset": "cifar10",
5
+ "batch_size": 64,
6
+ "epochs": 10,
7
+ "validation_frequency": 1,
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.0
16
+ },
17
+ "scheduler": "CosineAnnealingLR",
18
+ "scheduler_kwargs": {
19
+ "T_max": 10
20
+ },
21
+ "debug": false
22
+ }