supervised/config.yaml


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

codename: cifar10-resnet50-256-adam-linear
seed: -1
num_iters: 1000
log_dir: logs
checkpoint_dir: checkpoints

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