Switch to unified view

a b/configs/gaitset/gaitset.yaml
1
data_cfg:
2
  dataset_name: CASIA-B
3
  dataset_root: your_path
4
  num_workers: 1
5
  remove_no_gallery: false
6
  test_dataset_name: CASIA-B
7
8
evaluator_cfg:
9
  enable_float16: false
10
  restore_ckpt_strict: true
11
  restore_hint: 40000
12
  save_name: GaitSet
13
  sampler:
14
    batch_size: 16
15
    sample_type: all_ordered
16
    type: InferenceSampler
17
  metric: euc # cos
18
19
loss_cfg:
20
  loss_term_weight: 1.0
21
  margin: 0.2
22
  type: TripletLoss
23
  log_prefix: triplet
24
25
model_cfg:
26
  model: GaitSet
27
  in_channels:
28
    - 1
29
    - 32
30
    - 64
31
    - 128
32
  SeparateFCs:
33
    in_channels: 128
34
    out_channels: 256
35
    parts_num: 62
36
  bin_num:
37
    - 16
38
    - 8
39
    - 4
40
    - 2
41
    - 1
42
43
optimizer_cfg:
44
  lr: 0.1
45
  momentum: 0.9
46
  solver: SGD
47
  weight_decay: 0.0005
48
49
scheduler_cfg:
50
  gamma: 0.1
51
  milestones:
52
    - 10000
53
    - 20000
54
    - 30000
55
  scheduler: MultiStepLR
56
57
trainer_cfg:
58
  enable_float16: true
59
  log_iter: 100
60
  with_test: true
61
  restore_ckpt_strict: true
62
  restore_hint: 0
63
  save_iter: 10000
64
  save_name: GaitSet
65
  sync_BN: false
66
  total_iter: 40000
67
  sampler:
68
    batch_shuffle: false
69
    batch_size:
70
      - 8
71
      - 16
72
    frames_num_fixed: 30
73
    frames_num_max: 50
74
    frames_num_min: 25
75
    sample_type: fixed_unordered
76
    type: TripletSampler