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 |