From 693ebba3bd2967ba1528670eb68b6ee0a16f1b79 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Fri, 12 Aug 2022 11:41:34 +0800 Subject: Add SimCLR config --- simclr/config.example.yaml | 37 +++++++++++++++++++++++++++++++++++++ 1 file changed, 37 insertions(+) create mode 100644 simclr/config.example.yaml diff --git a/simclr/config.example.yaml b/simclr/config.example.yaml new file mode 100644 index 0000000..20bed04 --- /dev/null +++ b/simclr/config.example.yaml @@ -0,0 +1,37 @@ +codename: cifar10-simclr-128-lars-warmup-example +seed: -1 +num_iters: 19531 +log_dir: logs +checkpoint_dir: checkpoints + +hid_dim: 2048 +out_dim: 128 +temp: 0.5 + +dataset: cifar10 +dataset_dir: dataset +crop_size: 32 +crop_scale_range: + - 0.8 + - 1 +hflip_prob: 0.5 +distort_strength: 0.5 +#gauss_ker_scale: 10 +#gauss_sigma_range: +# - 0.1 +# - 2 +#gauss_prob: 0.5 + +batch_size: 128 +num_workers: 2 + +optim: lars +lr: 1 +momentum: 0.9 +#betas: +# - 0.9 +# - 0.999 +weight_decay: 1.0e-06 + +sched: warmup-anneal +warmup_iters: 1953 \ No newline at end of file -- cgit v1.2.3