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