-
Notifications
You must be signed in to change notification settings - Fork 168
/
gaitgraph2_OUMVLP17.yaml
108 lines (100 loc) · 2.06 KB
/
gaitgraph2_OUMVLP17.yaml
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
data_cfg:
dataset_name: OUMVLP
dataset_root: your_path
dataset_partition: ./datasets/OUMVLP/OUMVLP.json
test_dataset_name: OUMVLP
num_workers: 8
remove_no_gallery: false
evaluator_cfg:
enable_float16: false
restore_ckpt_strict: true
restore_hint: 80000
save_name: GaitGraph2_phase2_17
sampler:
batch_size: 256
frames_num_fixed: 30
frames_num_max: 50
frames_num_min: 25
sample_type: fixed_ordered
frames_skip_num: 0
metric: cos
eval_func: evaluate_indoor_dataset
transform:
- type: Compose
trf_cfg:
- type: NormalizeEmpty
- type: GaitGraphMultiInput
joint_format: coco
loss_cfg:
- loss_term_weight: 1
temperature: 0.01
type: SupConLoss_Lp
log_prefix: SupConLoss
model_cfg:
model: GaitGraph2
joint_format: coco
input_num: 3
reduction: 4
block: Bottleneck # Basic, initial
input_branch:
- 5
- 64
- 64
- 32
main_stream:
- 32
- 128
- 128
- 128
- 256
- 256
- 256
num_class: 128
tta: true
optimizer_cfg:
lr: 0.005
solver: Adam
weight_decay: 0.00001
scheduler_cfg:
three_phase: True
max_lr: 0.005
total_steps: 80000
scheduler: OneCycleLR
trainer_cfg:
enable_float16: false
log_iter: 1000
with_test: true
restore_ckpt_strict: false
restore_hint: 0
save_iter: 10000
save_name: GaitGraph2_phase2_17
sync_BN: true
total_iter: 80000
sampler:
batch_shuffle: true
frames_num_fixed: 30
frames_num_max: 50
frames_num_min: 25
sample_type: fixed_ordered #Repeat sample
frames_skip_num: 0
batch_size: 768
type: CommonSampler
transform:
- type: Compose
trf_cfg:
- type: NormalizeEmpty
- type: FlipSequence
probability: 0.5
- type: InversePosesPre
probability: 0.1
joint_format: coco
- type: JointNoise
std: 0.1
- type: PointNoise
std: 0.05
- type: RandomMove
random_r:
- 3
- 1
- type: GaitGraphMultiInput
joint_format: coco