| { |
| "adam_beta2": 0.999, |
| "arch": "trans_enc", |
| "autoregressive": false, |
| "autoregressive_include_prefix": false, |
| "autoregressive_init": "data", |
| "avg_model_beta": 0.9999, |
| "batch_size": 16, |
| "cond_mask_prob": 0.1, |
| "context_len": 0, |
| "cuda": true, |
| "data_dir": "", |
| "dataset": "humanml", |
| "device": 0, |
| "diffusion_steps": 50, |
| "emb_trans_dec": false, |
| "eval_batch_size": 32, |
| "eval_during_training": true, |
| "eval_metrics": [ |
| "fid", |
| "precision" |
| ], |
| "eval_num_samples": 1000, |
| "eval_rep_times": 3, |
| "eval_split": "val", |
| "eval_steps": [ |
| 50000, |
| 100000, |
| 200000 |
| ], |
| "external_mode": false, |
| "gen_during_training": false, |
| "gen_guidance_param": 2.5, |
| "gen_num_repetitions": 2, |
| "gen_num_samples": 3, |
| "lambda_fc": 0.0, |
| "lambda_layersync": 0.2, |
| "lambda_rcxyz": 0.0, |
| "lambda_target_loc": 0.0, |
| "lambda_vel": 0.0, |
| "latent_dim": 512, |
| "layers": 8, |
| "layersync_strong_layer": 6, |
| "layersync_weak_layer": 3, |
| "log_interval": 50, |
| "lr": 0.0001, |
| "lr_anneal_steps": 0, |
| "mask_frames": true, |
| "multi_encoder_type": "single", |
| "multi_target_cond": false, |
| "noise_schedule": "cosine", |
| "num_frames": 60, |
| "num_steps": 200000, |
| "overwrite": true, |
| "pos_embed_max_len": 5000, |
| "pred_len": 0, |
| "resume_checkpoint": "", |
| "save_dir": "./save/humanml_mdm_layersync_50steps_200k", |
| "save_interval": 50000, |
| "seed": 10, |
| "sigma_small": true, |
| "target_enc_layers": 1, |
| "target_joint_names": "DIMP_FINAL", |
| "text_encoder_type": "clip", |
| "train_platform_type": "WandBPlatform", |
| "unconstrained": false, |
| "use_ema": false, |
| "weight_decay": 0.0 |
| } |
