Update SATA release checkpoints

#1
by cyz020403 - opened
result/vae_animo/config.yaml ADDED
@@ -0,0 +1,91 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model:
2
+ Decoder:
3
+ attn_kwargs:
4
+ dropout: 0.1
5
+ attn_type: multihead
6
+ heads: 4
7
+ hidden_dim: 256
8
+ max_seq_len: 512
9
+ num_layers: 3
10
+ pos_encoding_type: sinusoidal
11
+ temporal_dropout: 0.1
12
+ temporal_ff_dim_multiplier: 2
13
+ temporal_num_heads: 4
14
+ temporal_type: transformer
15
+ type: GPSDecTemporal
16
+ use_temporal_pos_encoding: true
17
+ Encoder:
18
+ attn_kwargs:
19
+ dropout: 0.1
20
+ attn_type: multihead
21
+ heads: 4
22
+ hidden_dim: 256
23
+ max_seq_len: 512
24
+ num_layers: 3
25
+ pos_encoding_type: sinusoidal
26
+ sample_posterior: true
27
+ temporal_dropout: 0.1
28
+ temporal_ff_dim_multiplier: 2
29
+ temporal_num_heads: 4
30
+ temporal_type: transformer
31
+ type: GPSEncTemporal
32
+ use_temporal_pos_encoding: true
33
+ z_dim: 128
34
+ representation:
35
+ out:
36
+ - r
37
+ - q
38
+ - c
39
+ pose:
40
+ - q
41
+ - p
42
+ - r
43
+ - pv
44
+ - qv
45
+ - c
46
+ skel:
47
+ - lo
48
+ - go
49
+ - tf
50
+ test_data:
51
+ dir: zooV2_test_wtextV2/motion/processed
52
+ sample_num: 50
53
+ train:
54
+ batch_size: 12
55
+ consq_n: 64
56
+ copy_orig_contact: false
57
+ epoch_num: 400
58
+ grad_max_norm: 0.5
59
+ learning_rate: 0.0003
60
+ load:
61
+ dir: same_vae_lowerLR_512_wtextV2_nomask_newLoss_gps_t64_temporal_trans_trainZooV2
62
+ epoch: 295
63
+ loss:
64
+ c: 1
65
+ cv: 6
66
+ jerk: 0.2
67
+ p: 0.01
68
+ pen: 0.1
69
+ pv: 1
70
+ q: 5
71
+ r: 10
72
+ slide: 6
73
+ lr_schedule:
74
+ gamma: 0.99
75
+ min: 0.01
76
+ metric:
77
+ - qR
78
+ - ra_xz
79
+ - pa
80
+ - slide
81
+ - jerk
82
+ - pen
83
+ min_motion_lens: 20
84
+ num_workers: 2
85
+ save_per: 5
86
+ vae_kl:
87
+ kl_weight: 1.0e-06
88
+ nll_loss_type: null
89
+ train_data:
90
+ dir: zooV2_train_wtextV2/motion/processed
91
+ mask: []
result/vae_animo/last_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e298455a8bcc9728e5d7c427b0d1f5f40520f806f60d73d0e3b8cf1fa072d7e6
3
+ size 101249914
result/vae_animo/ms_dict.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a090f00994c89ee9f4618c89f988df232773ef66c621b74d51d40859525b4ec3
3
+ size 4902
result/vae_human/{model_390.pt → last_model.pt} RENAMED
File without changes
result/vae_merge/last_model.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ce79b15dfaa3b3b1958b9f3c8241295bd0deb9dfcec8a5ce6ee9426c54c6bbf4
3
- size 101248332
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6847ad2d36b78d64ff83cbcdda70701b1f405c9061fb04609ae8f4f28a281c06
3
+ size 101247418
src/momask_same_reprod/checkpoints/t2m/evaluator_meta/meta/mean.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0bdb5ba69a3a9e34d71990db15bc535ebc024c8d95ddb5574196f96058faa7d3
3
+ size 2232
src/momask_same_reprod/checkpoints/t2m/evaluator_meta/meta/std.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a5f7d60301c9465972fc225f8ad0ee8f957e7720431189123eb6d15873a9557
3
+ size 2232
src/momask_same_reprod/checkpoints/t2m/text_mot_match/model/finest.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:881e45bea6fb84b15eca6e7cdd75d602b58a72f2038c223ee96ab75d4f684ada
3
+ size 245580211