base_config:
  - egs/egs_bases/audio2motion/vae_sync.yaml

task_cls: tasks.postnet.lm3d_postnet_adv_sync.PostnetAdvSyncTask
audio2motion_task_cls: tasks.audio2motion.lm3d_vae_sync.VAESyncAudio2MotionTask
person_binary_data_dir: data/binary/videos
# postnet training
postnet_lr: 0.0001
postnet_lambda_adv: 0.85
postnet_lambda_sync: 0.1
postnet_lambda_mse: 0.05

# Discriminator
postnet_disc_lr: 0.0001
discriminator_scheduler_params:
  gamma: 0.5
  step_size: 40000
postnet_disc_start_steps: 0
postnet_disc_interval: 1

# Training Schedule
scheduler: none
num_ckpt_keep: 500
val_check_interval: 1000
valid_infer_interval: 1000
max_updates: 100000 # 20000

# Pretrained Ckpts
audio2motion_work_dir: checkpoints/th1kh/lm3d_vae_sync_pitch/
audio2motion_ckpt_steps: 160000
syncnet_work_dir: checkpoints/th1kh/lm3d_syncnet
syncnet_ckpt_steps: 160000
syncnet_num_layers_per_block: 3
syncnet_base_hid_size: 128

infer_audio_source_name: data/raw/val_wavs/zozo.wav
infer_out_npy_name: infer_out/May/pred_lm3d/zozo.npy
infer_ckpt_steps: 6000

load_db_to_memory: false # enable it for faster indexing