1 files changed, 15 insertions, 0 deletions
diff --git a/configs/cifar10.yaml b/configs/cifar10.yaml
new file mode 100644
index 0000000..6429287
--- /dev/null
+++ b/configs/cifar10.yaml
@@ -0,0 +1,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