Switch to unified view

a b/configs/gaitpart/gaitpart.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: 120000
12
  save_name: GaitPart
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: GaitPart
27
  backbone_cfg:
28
    in_channels: 1
29
    layers_cfg:
30
      - BC-32
31
      - BC-32
32
      - M
33
      - FC-64-2
34
      - FC-64-2
35
      - M
36
      - FC-128-3
37
      - FC-128-3
38
    type: Plain
39
  SeparateFCs:
40
    in_channels: 128
41
    out_channels: 128
42
    parts_num: 16
43
  bin_num:
44
    - 16
45
46
47
optimizer_cfg:
48
  lr: 0.0001
49
  momentum: 0.9
50
  solver: Adam
51
  weight_decay: 0.0
52
53
scheduler_cfg:
54
  gamma: 0.1
55
  milestones:
56
    - 100000
57
  scheduler: MultiStepLR
58
59
trainer_cfg:
60
  enable_float16: true
61
  log_iter: 100
62
  with_test: true
63
  restore_ckpt_strict: true
64
  restore_hint: 0
65
  save_iter: 10000
66
  save_name: GaitPart
67
  sync_BN: false
68
  total_iter: 120000
69
  sampler:
70
    batch_shuffle: false
71
    batch_size:
72
      - 8
73
      - 16
74
    frames_num_fixed: 30
75
    frames_num_max: 50
76
    frames_num_min: 25
77
    frames_skip_num: 10
78
    sample_type: fixed_ordered
79
    type: TripletSampler