Update SATA release checkpoints
#1
by cyz020403 - opened
- result/vae_animo/config.yaml +91 -0
- result/vae_animo/last_model.pt +3 -0
- result/vae_animo/ms_dict.pt +3 -0
- result/vae_human/{model_390.pt → last_model.pt} +0 -0
- result/vae_merge/last_model.pt +2 -2
- src/momask_same_reprod/checkpoints/t2m/evaluator_meta/meta/mean.npy +3 -0
- src/momask_same_reprod/checkpoints/t2m/evaluator_meta/meta/std.npy +3 -0
- src/momask_same_reprod/checkpoints/t2m/text_mot_match/model/finest.tar +3 -0
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:
|
| 3 |
-
size
|
|
|
|
| 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
|