mirror of https://github.com/sthalles/SimCLR.git
22 lines
317 B
YAML
22 lines
317 B
YAML
batch_size: 512
|
|
epochs: 80
|
|
eval_every_n_epochs: 1
|
|
fine_tune_from: None
|
|
log_every_n_steps: 50
|
|
weight_decay: 10e-6
|
|
fp16_precision: False
|
|
|
|
model:
|
|
out_dim: 256
|
|
base_model: "resnet18"
|
|
|
|
dataset:
|
|
s: 1
|
|
input_shape: (96,96,3)
|
|
num_workers: 0
|
|
valid_size: 0.05
|
|
|
|
loss:
|
|
temperature: 0.5
|
|
use_cosine_similarity: True
|