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 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) create mode 100644 supervised/config-resnet.yaml (limited to 'supervised/config-resnet.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 -- cgit v1.2.3