From 3d57389a007e195415c4e7708d5a7a74860c194f Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Fri, 12 Aug 2022 17:04:23 +0800
Subject: Add SimCLR evaluation config

---
 simclr/config_eval.example.yaml | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 simclr/config_eval.example.yaml

(limited to 'simclr')

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
-- 
cgit v1.2.3