From 81597cdd0a55140f50b32b69507bfa5309b75f44 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Wed, 10 Aug 2022 18:04:50 +0800 Subject: Update config files --- supervised/config-resnet.yaml | 29 +++++++++++++++++++++++++++++ supervised/config-vit.yaml | 6 +++--- supervised/config.yaml | 28 ---------------------------- 3 files changed, 32 insertions(+), 31 deletions(-) create mode 100644 supervised/config-resnet.yaml delete mode 100644 supervised/config.yaml 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 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 diff --git a/supervised/config.yaml b/supervised/config.yaml deleted file mode 100644 index a3cf4f6..0000000 --- a/supervised/config.yaml +++ /dev/null @@ -1,28 +0,0 @@ -codename: cifar10-resnet50-256-adam-linear -seed: -1 -num_iters: 1000 -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: 256 -num_workers: 2 - -optim: adam -lr: 0.001 -betas: - - 0.9 - - 0.999 -weight_decay: 1.0e-06 - -sched: linear \ No newline at end of file -- cgit v1.2.3