arcface-pytorch/config.yaml at master · happydog-intj/arcface-pytorch · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
# Optimized Configuration for Better Training
model:
  backbone: 'resnet18'
  metric: 'arc_margin'
  use_se: false
  easy_margin: false
  num_classes: 100

loss:
  type: 'focal_loss'
  gamma: 2

training:
  batch_size: 32
  test_batch_size: 16
  max_epoch: 100
  lr: 0.001
  lr_step: 20
  weight_decay: 0.0005
  optimizer: 'sgd'
  num_workers: 0
  print_freq: 10
  save_interval: 10

data:
  train_root: './data/datasets/lfw/lfw-align-128'
  train_list: './train_list.txt'
  lfw_root: './data/datasets/lfw/lfw-align-128'
  lfw_test_list: './lfw_test_pair.txt'
  input_shape: [1, 128, 128]

paths:
  checkpoints_path: 'checkpoints'
  test_model_path: 'checkpoints/resnet18_best.pth'

system:
  use_gpu: false
  gpu_id: '0'
  display: false