Hugging Face's logo

SkeletonDiffusion
/
ModelCheckpoints

English
human-motion-generation
human-motion-prediction
probabilistic-human-motion-generation
Model card Files
xet
Community
ModelCheckpoints
File size: 2,242 Bytes 
98f391d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
109
110
111
112
113
114
115
 
dataset_main_path: ./datasets
dataset_annotation_path: ./datasets/annotations
dataset_precomputed_path: ./datasets/processed
resume_training: false
device: cuda
load: false
load_path: ''
output_log_path: ./trained_models/hmp/amass/autoencoder
batch_size: 64
batch_size_eval: 512
if_run_validation: False
eval_frequency: 25
num_epochs: 300
num_iteration_eval: 10
num_workers: 4
seed: 52345
use_lr_scheduler: true
lr_scheduler_kwargs:
  lr_scheduler_type: ExponentialLRSchedulerWarmup
  warmup_duration: 10
  update_every: 1
  min_lr: 0.0001
  gamma_decay: 0.98
loss_pose_type: l1
lr: 0.005
latent_size: 96
output_size: 3
z_activation: tanh
num_iter_perepoch: 580
obs_length: 30
prediction_horizon_train: 120
prediction_horizon_eval: 120
prediction_horizon: 120
pred_length: 120
prediction_horizon_train_min: 10
prediction_horizon_train_min_from_epoch: 200
curriculum_it: 10
random_prediction_horizon: true
if_consider_hip: false
history_sec: 0.5
prediction_horizon_sec: 2
motion_repr_type: SkeletonRescalePose
pose_box_size: 1.2
seq_centering: 0
task_name: hmp
num_joints: 22
fps: 60
multimodal_threshold: 0.4
dataset_type: AMASSDataset
dataset_name: amass
precomputed_folder: ./datasets/processed/AMASS/hmp
annotations_folder: ./datasets/annotations/AMASS/hmp
dtype: float32
data_loader_train:
  stride: 60
  augmentation: 30
  shuffle: true
  datasets:
  - ACCAD
  - BMLhandball
  - BMLmovi
  - BMLrub
  - EKUT
  - CMU
  - EyesJapanDataset
  - KIT
  - PosePrior
  - TCDHands
  - TotalCapture
  da_mirroring: 0.5
  da_rotations: 1.0
  drop_last: true
  if_load_mmgt: false
data_loader_train_eval:
  stride: 30
  augmentation: 0
  shuffle: false
  datasets:
  - ACCAD
  - BMLhandball
  - BMLmovi
  - BMLrub
  - EKUT
  - CMU
  - EyesJapanDataset
  - KIT
  - PosePrior
  - TCDHands
  - TotalCapture
  da_mirroring: 0.0
  da_rotations: 0.0
  drop_last: false
  if_load_mmgt: false
data_loader_valid:
  stride: 30
  augmentation: 0
  shuffle: false
  datasets:
  - HumanEva
  - HDM05
  - SFU
  - MoSh
  file_idces: all
  drop_last: false
  if_load_mmgt: false

autoenc_arch:
  enc_num_layers: 2
  encoder_hidden_size: 96
  decoder_hidden_size: 96
  arch: AutoEncoder 
  recurrent_arch_enc: StaticGraphGRU
  recurrent_arch_decoder: StaticGraphGRU