diff options
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 |