feature_extractor/config.yaml

batch_size: 256
epochs: 20
eval_every_n_epochs: 1
fine_tune_from: ''
log_every_n_steps: 25
weight_decay: 10e-6
fp16_precision: False
n_gpu: 2
gpu_ids: (0,1)

model:
  out_dim: 512
  base_model: "resnet18"

dataset:
  s: 1
  input_shape: (224,224,3)
  num_workers: 10
  valid_size: 0.1

loss:
  temperature: 0.5
  use_cosine_similarity: True