| 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/freeman/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: 580 | |
| obs_length: 15 | |
| prediction_horizon_train: 60 | |
| prediction_horizon_eval: 60 | |
| prediction_horizon: 60 | |
| pred_length: 60 | |
| 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: 18 | |
| fps: 30 | |
| multimodal_threshold: 0.5 | |
| dataset_type: FreeManDataset | |
| dataset_name: freeman | |
| precomputed_folder: ./datasets/processed/FreeMan/hmp | |
| annotations_folder: ./datasets/annotations/FreeMan/hmp | |
| dtype: float32 | |
| data_loader_train: | |
| stride: 10 | |
| augmentation: 5 | |
| shuffle: true | |
| 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 | |
| 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 | |
| segments_path: ./datasets/annotations/FreeMan/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 |