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 | |
| parent | 95d5e4e82df0de08210088d07d6d89692f1325d1 (diff) | |
Update config files
Diffstat (limited to 'supervised')
| -rw-r--r-- | supervised/config-resnet.yaml (renamed from supervised/config.yaml) | 9 | ||||
| -rw-r--r-- | supervised/config-vit.yaml | 6 |
2 files changed, 8 insertions, 7 deletions
diff --git a/supervised/config.yaml b/supervised/config-resnet.yaml index a3cf4f6..e51b364 100644 --- a/supervised/config.yaml +++ b/supervised/config-resnet.yaml @@ -1,6 +1,6 @@ -codename: cifar10-resnet50-256-adam-linear +codename: cifar10-resnet-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 @@ -25,4 +25,5 @@ betas: - 0.999 weight_decay: 1.0e-06 -sched: linear
\ No newline at end of file +sched: warmup-anneal +warmup_iters: 5
\ No newline at end of file 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 |