File size: 341 Bytes
d60982d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
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
|