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