From d01affeafed02701f128a06899fb658441475792 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Thu, 18 Aug 2022 12:07:30 +0800 Subject: Add SimCLR ViT variant for CIFAR --- simclr/config.example.yaml | 1 + 1 file changed, 1 insertion(+) (limited to 'simclr/config.example.yaml') diff --git a/simclr/config.example.yaml b/simclr/config.example.yaml index 20bed04..ea9b9e3 100644 --- a/simclr/config.example.yaml +++ b/simclr/config.example.yaml @@ -4,6 +4,7 @@ num_iters: 19531 log_dir: logs checkpoint_dir: checkpoints +encoder: resnet hid_dim: 2048 out_dim: 128 temp: 0.5 -- cgit v1.2.3