| 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 | |