diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-10 18:04:50 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-10 18:04:50 +0800 |
| commit | 81597cdd0a55140f50b32b69507bfa5309b75f44 (patch) | |
| tree | 72db21200f67a0a2b59bcf5a5a1f0c136d0daf32 /supervised/config-vit.yaml | |
| parent | 95d5e4e82df0de08210088d07d6d89692f1325d1 (diff) | |
Update config files
Diffstat (limited to 'supervised/config-vit.yaml')
| -rw-r--r-- | supervised/config-vit.yaml | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/supervised/config-vit.yaml b/supervised/config-vit.yaml index 5ed6689..1b1ded7 100644 --- a/supervised/config-vit.yaml +++ b/supervised/config-vit.yaml @@ -1,6 +1,6 @@ -codename: cifar10-vit-adam-warmup-anneal +codename: cifar10-vit-128-adam-warmup-anneal seed: -1 -num_iters: 1000 +num_iters: 200 log_dir: logs checkpoint_dir: checkpoints @@ -15,7 +15,7 @@ crop_scale_range: - 1 hflip_prob: 0.5 -batch_size: 256 +batch_size: 128 num_workers: 2 optim: adam |