-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathjoint.yaml
More file actions
50 lines (45 loc) · 849 Bytes
/
joint.yaml
File metadata and controls
50 lines (45 loc) · 849 Bytes
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
work_dir: ./work_dir/ntu60/xsub/ctrgcn_joint
# feeder
feeder: feeders.feeder_ntu.Feeder
train_feeder_args:
data_path: data/ntu/NTU60_CS.npz
split: train
debug: False
random_choose: False
random_shift: False
random_move: False
window_size: 64
normalization: False
random_rot: True
p_interval: [0.5, 1]
vel: False
bone: False
test_feeder_args:
data_path: data/ntu/NTU60_CS.npz
split: test
window_size: 64
p_interval: [0.95]
vel: False
bone: False
debug: False
# model
model: model.ctrgcn.Model
model_args:
num_class: 60
num_point: 25
num_person: 2
graph: graph.ntu_rgb_d.Graph
graph_args:
labeling_mode: 'spatial'
#optim
weight_decay: 0.0004
base_lr: 0.1
lr_decay_rate: 0.1
step: [35, 55]
warm_up_epoch: 5
# training
#device: 0
batch_size: 512
test_batch_size: 512
num_epoch: 65
nesterov: True