{
  "global_step": 200000,
  "checkpoint_file": "step_0200000_final.pt",
  "training_args": {
    "data_root": "/home/signvrse/RVQ-VAE/data/signvrse_rvqvae_motion_dataset",
    "checkpoint_dir": "/home/signvrse/RVQ-VAE/checkpoints",
    "resume": null,
    "input_dim": 668,
    "latent_dim": 512,
    "num_codebooks": 6,
    "codebook_size": 512,
    "quant_dropout": 0.2,
    "ema_decay": 0.99,
    "seq_len": 64,
    "window_stride": 32,
    "num_workers": 4,
    "batch_size": 128,
    "grad_accum_steps": 4,
    "lr": 0.0002,
    "warmup_steps": 2000,
    "max_steps": 200000,
    "velocity_alpha": 0.5,
    "commitment_beta": 0.02,
    "grad_clip": 1.0,
    "log_every": 50,
    "val_every": 2000,
    "save_every": 5000,
    "wandb_project": "rvq-vae-momask",
    "wandb_run_name": null,
    "device": null
  }
}