model:
  name: 'resnet50_fc512'
  pretrained: True
  num_train_classes: 384

data:
  type: 'image'
  source: 'wildhorse_face'
  height: 300
  width: 150
  transforms_train: ['resize', 'random_crop', 'random_affine']
  transforms_test: ['resize']
  version: 'v2'

loss:
  name: 'triplet'
  triplet:
    weight_t: 1.
    weight_x: 1.
  softmax:
    label_smooth: True

sampler:
  num_instances: 3
  num_copies: 3

train:
  optim: 'adam'
  lr: 0.00001
  max_epoch: 1000
  batch_size: 64
  fixbase_epoch: 1
  open_layers: ['fc', 'classifier']
  lr_scheduler: 'multi_step'
  stepsize: [400]

test:
  batch_size: 100
  dist_metric: 'euclidean'
  normalize_feature: False
  evaluate: False
  eval_freq: 10
  rerank: False
  ranks: [1, 5, 12, 20]
  visrank_topk: 12
  visrank_resize: False
