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