diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-12 17:04:23 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-12 17:04:23 +0800 |
| commit | 3d57389a007e195415c4e7708d5a7a74860c194f (patch) | |
| tree | ca890b3a2191d971d50439569741f6721a140f5e /simclr/config_eval.example.yaml | |
| parent | 693ebba3bd2967ba1528670eb68b6ee0a16f1b79 (diff) | |
Add SimCLR evaluation config
Diffstat (limited to 'simclr/config_eval.example.yaml')
| -rw-r--r-- | simclr/config_eval.example.yaml | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/simclr/config_eval.example.yaml b/simclr/config_eval.example.yaml new file mode 100644 index 0000000..753926b --- /dev/null +++ b/simclr/config_eval.example.yaml @@ -0,0 +1,29 @@ +codename: cifar10-simclr-128-lars-warmup-linear-example +seed: -1 +num_iters: 90 +log_dir: logs +checkpoint_dir: checkpoints + +hid_dim: 2048 +out_dim: 128 + +dataset: cifar10 +dataset_dir: dataset +train_size: 32 +test_size: 32 +test_crop_size: 32 +hflip_prob: 0.5 + +batch_size: 256 +num_workers: 2 + +optim: sgd +lr: 0.1 +momentum: 0.9 +#betas: +# - 0.9 +# - 0.999 +weight_decay: 0.0 + +sched: const +#warmup_iters: 5
\ No newline at end of file |