aboutsummaryrefslogtreecommitdiff
path: root/supervised/config-resnet.yaml
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2022-08-10 18:04:50 +0800
committerJordan Gong <jordan.gong@protonmail.com>2022-08-10 18:04:50 +0800
commit81597cdd0a55140f50b32b69507bfa5309b75f44 (patch)
tree72db21200f67a0a2b59bcf5a5a1f0c136d0daf32 /supervised/config-resnet.yaml
parent95d5e4e82df0de08210088d07d6d89692f1325d1 (diff)
Update config files
Diffstat (limited to 'supervised/config-resnet.yaml')
-rw-r--r--supervised/config-resnet.yaml29
1 files changed, 29 insertions, 0 deletions
diff --git a/supervised/config-resnet.yaml b/supervised/config-resnet.yaml
new file mode 100644
index 0000000..e51b364
--- /dev/null
+++ b/supervised/config-resnet.yaml
@@ -0,0 +1,29 @@
+codename: cifar10-resnet-128-adam-warmup-anneal
+seed: -1
+num_iters: 200
+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: 128
+num_workers: 2
+
+optim: adam
+lr: 0.001
+betas:
+ - 0.9
+ - 0.999
+weight_decay: 1.0e-06
+
+sched: warmup-anneal
+warmup_iters: 5 \ No newline at end of file