| { | |
| "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 | |
| } |