diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-18 12:07:30 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-18 12:07:30 +0800 |
| commit | d01affeafed02701f128a06899fb658441475792 (patch) | |
| tree | 5c41b549a7595afe0fde70962f348a834e5d0b3e /simclr/config.example.yaml | |
| parent | 1b8f01ce9706905c36c6f11ed9deac8548ad7341 (diff) | |
Add SimCLR ViT variant for CIFAR
Diffstat (limited to 'simclr/config.example.yaml')
| -rw-r--r-- | simclr/config.example.yaml | 1 |
1 files changed, 1 insertions, 0 deletions
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 |