args.json

{
    "batch_size": 64,
    "bpe_training_ratio": 0.5,
    "cond_mask_prob": 0.1,
    "cuda": true,
    "data_dir": "",
    "dataset": "humanml",
    "device": 0,
    "diffusion_steps": 1000,
    "lambda_fc": 0.0,
    "lambda_rcxyz": 0.0,
    "lambda_vel": 0.0,
    "lambda_vel_rcxyz": 0.0,
    "latent_dim": 512,
    "layers": 8,
    "log_interval": 1000,
    "lr": 0.0001,
    "lr_anneal_steps": 0,
    "max_mod": 10,
    "max_seq_att": 1024,
    "max_seq_len": 200,
    "min_seq_len": 70,
    "noise_schedule": "cosine",
    "num_frames": 60,
    "num_heads": 4,
    "num_steps": 600000,
    "num_workers": 4,
    "overwrite": true,
    "pose_rep": "rotvec",
    "protocol": "",
    "resume_checkpoint": "",
    "rpe_horizon": 150,
    "save_dir": "results/humanml/FlowMDM",
    "save_interval": 50000,
    "seed": 10,
    "sigma_small": true,
    "train_platform_type": "WandbPlatform",
    "unconstrained": false,
    "weight_decay": 0.0
}