forked from davrempe/humor
-
Notifications
You must be signed in to change notification settings - Fork 1
/
train_humor_qual.cfg
55 lines (46 loc) · 1.03 KB
/
train_humor_qual.cfg
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
--dataset AmassDiscreteDataset
--data-paths ./data/amass_processed
--split-by dataset
--sample-num-frames 10
--data-steps-in 1
--data-steps-out 1
--data-rot-rep mat
--data-return-config smpl+joints+contacts
--model HumorModel
--model-data-config smpl+joints+contacts
--in-rot-rep mat
--out-rot-rep aa
--latent-size 48
--model-steps-in 1
--model-use-smpl-joint-inputs
--loss HumorLoss
--kl-loss 0.0004
--kl-loss-anneal-start 0
--kl-loss-anneal-end 50
--contacts-loss 0.01
--regr-trans-loss 1.0
--regr-trans-vel-loss 1.0
--regr-root-orient-loss 1.0
--regr-root-orient-vel-loss 1.0
--regr-pose-loss 1.0
--regr-pose-vel-loss 1.0
--regr-joint-loss 1.0
--regr-joint-vel-loss 1.0
--regr-joint-orient-vel-loss 1.0
--regr-vert-loss 1.0
--regr-vert-vel-loss 1.0
--smpl-joint-loss 1.0
--smpl-mesh-loss 1.0
--smpl-joint-consistency-loss 1.0
--out ./out/humor_qual_train
--gpu 0
--batch-size 200
--epochs 200
--lr 1e-4
--sched-milestones 50 80 140
--sched-decay 0.5 0.2 0.4
--sched-samp-start 10
--sched-samp-end 20
--val-every 2
--save-every 25
--print-every 10