configs/cifar10.yaml


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

dataset: cifar10
d_hidden: 512
num_blocks: 12
batch_size: 128
epochs: 100
lr: 0.001
lr_fb: 0.001
wd: 0.01
lam: 0.1
K: 4
sigma_bridge: 0.03
ema_momentum: 0.995
seeds: [42, 123, 456]
gpu: 1
output_dir: results/cifar10