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,106 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
 
dataset_main_path: ./datasets
dataset_annotation_path: ./datasets/annotations
dataset_precomputed_path: ./datasets/processed
if_resume_training: false
device: cuda
load: false
load_path: ''
output_log_path: ./trained_models/hmp/h36m/autoencoder
batch_size: 64
batch_size_eval: 512
if_run_validation: false
eval_frequency: 25
num_epochs: 200
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: 485
obs_length: 25
prediction_horizon_train: 100
prediction_horizon_eval: 100
prediction_horizon: 100
pred_length: 100
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.5
seq_centering: 0
task_name: hmp
num_joints: 17
fps: 50
multimodal_threshold: 0.5
dataset_type: H36MDataset
dataset_name: h36m
precomputed_folder: ./datasets/processed/Human36M/hmp/
annotations_folder: ./datasets/annotations/Human36M/hmp
dtype: float32
data_loader_train:
  stride: 10
  augmentation: 5
  shuffle: true
  subjects:
  - S1
  - S5
  - S6
  - S7
  - S8
  actions: all
  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
  subjects:
  - S1
  - S5
  - S6
  - S7
  - S8
  actions: all
  da_mirroring: 0.0
  da_rotations: 0.0
  drop_last: false
  if_load_mmgt: false
data_loader_valid:
  stride: 30
  augmentation: 0
  shuffle: false
  subjects:
  - S8
  segments_path: ./datasets/annotations/Human36M/hmp/segments_valid.csv
  actions: all
  drop_last: false
  if_load_mmgt: false


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