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