aboutsummaryrefslogtreecommitdiff
path: root/supervised/config.yaml
blob: a3cf4f699d0faa1f36dd1eb6a5fc4a93f9d38ae7 (plain)
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
27
28
codename: cifar10-resnet50-256-adam-linear
seed: -1
num_iters: 1000
log_dir: logs
checkpoint_dir: checkpoints

backbone: resnet
label_smooth: 0

dataset: cifar10
dataset_dir: dataset
crop_size: 32
crop_scale_range:
  - 0.8
  - 1
hflip_prob: 0.5

batch_size: 256
num_workers: 2

optim: adam
lr: 0.001
betas:
  - 0.9
  - 0.999
weight_decay: 1.0e-06

sched: linear