Spaces:
Configuration error
Configuration error
# General | |
train_2d: True | |
no_eval: False | |
finetune: False | |
partial_train: null | |
# Traning | |
epochs: 90 | |
checkpoint_frequency: 30 | |
batch_size: 64 | |
dropout: 0.0 | |
learning_rate: 0.0005 | |
weight_decay: 0.01 | |
lr_decay: 0.99 | |
pretrain_3d_curriculum: 30 | |
# Model | |
maxlen: 243 | |
dim_feat: 512 | |
mlp_ratio: 2 | |
depth: 5 | |
dim_rep: 512 | |
num_heads: 8 | |
att_fuse: True | |
# Data | |
data_root: data/motion3d/MB3D_f243s81/ | |
subset_list: [AMASS, H36M-SH] | |
dt_file: h36m_sh_conf_cam_source_final.pkl | |
clip_len: 243 | |
data_stride: 81 | |
rootrel: True | |
sample_stride: 1 | |
num_joints: 17 | |
no_conf: False | |
gt_2d: False | |
# Loss | |
lambda_3d_velocity: 20.0 | |
lambda_scale: 0.5 | |
lambda_lv: 0.0 | |
lambda_lg: 0.0 | |
lambda_a: 0.0 | |
lambda_av: 0.0 | |
# Augmentation | |
synthetic: True # synthetic: don't use 2D detection results, fake it (from 3D) | |
flip: True | |
mask_ratio: 0.05 | |
mask_T_ratio: 0.1 | |
noise: True | |
noise_path: params/synthetic_noise.pth | |
d2c_params_path: params/d2c_params.pkl | |