diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-10 17:02:27 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-10 17:02:27 +0800 |
| commit | 95d5e4e82df0de08210088d07d6d89692f1325d1 (patch) | |
| tree | ed6b6788383ba77ff439afdc6da4490cb467eba9 /supervised/config-vit.yaml | |
| parent | a035bbdd0cc5e39954272d13d1e1595db738846a (diff) | |
Add supervised ViT baseline
Diffstat (limited to 'supervised/config-vit.yaml')
| -rw-r--r-- | supervised/config-vit.yaml | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/supervised/config-vit.yaml b/supervised/config-vit.yaml new file mode 100644 index 0000000..5ed6689 --- /dev/null +++ b/supervised/config-vit.yaml @@ -0,0 +1,29 @@ +codename: cifar10-vit-adam-warmup-anneal +seed: -1 +num_iters: 1000 +log_dir: logs +checkpoint_dir: checkpoints + +backbone: vit +label_smooth: 0.1 + +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: 5.0e-05 + +sched: warmup-anneal +warmup_iters: 5
\ No newline at end of file |