config.yaml

if_resume_training: false
debug: false
device: cuda
load: false
load_path: ''
dataset_main_path: ./datasets
dataset_annotation_path: ./datasets/annotations&interm
dataset_precomputed_path: ./datasets/processed
_load_saved_aoutoenc: hmp-amass
output_log_path: ./models/
slurm_id: 0
slurm_first_run: None
info: cvpr_release
pretrained_autoencoder_path: ./models/checkpoint_300.pt
lr: 0.001
diffusion_objective: pred_x0
weight_decay: 0.0
if_use_ema: true
step_start_ema: 100
ema_power: 0.6666666666666666
ema_update_every: 10
ema_min_value: 0.0
use_lr_scheduler: true
lr_scheduler_kwargs:
  lr_scheduler_type: ExponentialLRSchedulerWarmup
  warmup_duration: 75
  update_every: 2
  min_lr: 0.0001
  gamma_decay: 0.98
  decay_factor: 0.1
  patience: 1
diffusion_conditioning: true
num_epochs: 150
num_workers: 4
batch_size: 64
batch_size_eval: 256
eval_frequency: 1
train_pick_best_sample_among_k: 50
diffusion_k_train_after_epochs: 0
diffusion_k_train_monot_increase: false
diffusion_k_train_monot_increase_howmanyperepoch: 1
similarity_space: input_space
diffusion_activation: identity
num_prob_samples: 50
diffusion_timesteps: 10
diffusion_type: NonisotropicGaussianDiffusion
diffusion_loss_type: snr
loss_reduction_type: l1
if_run_as_isotropic: false
if_sigma_n_scale: true
diffusion_covariance_type: skeleton-diffusion
gamma_scheduler: cosine
beta_schedule: cosine
sigma_n_scale: spectral
num_iter_perepoch: null
seed: 63485
motion_repr_type: SkeletonRescalePose
dataset_type: AMASSDataset
dataset_name: amass
latent_size: 96
output_size: 3
z_activation: tanh
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
pose_box_size: 1.2
seq_centering: 0
task_name: hmp
num_joints: 22
fps: 60
multimodal_threshold: 0.4
precomputed_folder: ./models/
annotations_folder: ./models/
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
enc_num_layers: 2
encoder_hidden_size: 96
decoder_hidden_size: 96
arch: AutoEncoder
recurrent_arch_enc: StaticGraphGRU
recurrent_arch_decoder: StaticGraphGRU
diffusion_arch:
  arch: Denoiser
  use_attention: true
  self_condition: false
  norm_type: none
  depth: 4
  attn_dim_head: 32
  attn_heads: 8
  learn_influence: true
covariance_matrix_type: adjacency
hydra:
  run:
    dir: ./models
  job:
    chdir: true