SimCLR/config.yaml

9 lines
163 B
YAML

batch_size: 256
out_dim: 128
s: 1
temperature: 0.5
base_convnet: "resnet18" # one of: "resnet18 or resnet50"
use_cosine_similarity: True
epochs: 40
num_workers: 4