| { |
| "adam_beta2": 0.999, |
| "arch": "trans_dec", |
| "avg_model_beta": 0.9999, |
| "batch_size": 64, |
| "cond_mask_prob": 0.1, |
| "context_len": 20, |
| "cuda": true, |
| "data_dir": "", |
| "dataset": "humanml", |
| "device": 0, |
| "diffusion_steps": 10, |
| "emb_before_mask": false, |
| "emb_policy": "concat", |
| "emb_trans_dec": false, |
| "eval_batch_size": 32, |
| "eval_during_training": true, |
| "eval_num_samples": 1000, |
| "eval_rep_times": 3, |
| "eval_split": "test", |
| "gen_during_training": true, |
| "gen_guidance_param": 2.5, |
| "gen_num_repetitions": 1, |
| "gen_num_samples": 6, |
| "hml_type": null, |
| "keyframe_cond_prob": 0.5, |
| "keyframe_cond_type": "", |
| "lambda_fc": 0.0, |
| "lambda_rcxyz": 0.0, |
| "lambda_target_loc": 1.0, |
| "lambda_vel": 0.0, |
| "latent_dim": 512, |
| "layers": 8, |
| "log_interval": 1000, |
| "lr": 0.0001, |
| "lr_anneal_steps": 0, |
| "mask_frames": true, |
| "multi_encoder_type": "single", |
| "multi_target_cond": true, |
| "noise_schedule": "cosine", |
| "num_frames": 60, |
| "num_steps": 600000, |
| "overwrite": true, |
| "pos_embed_max_len": 5000, |
| "pred_len": 40, |
| "resume_checkpoint": "", |
| "sampling_mode": "none", |
| "save_dir": "save/MultiJointSearch_EmbPolicy_concat_MultiEncoderType_single_TargetEncLayers_1", |
| "save_interval": 50000, |
| "seed": 10, |
| "sigma_small": true, |
| "spatial_condition": null, |
| "target_enc_layers": 1, |
| "target_joint_names": "DIMP_FULL", |
| "text_encoder_type": "bert", |
| "train_platform_type": "WandBPlatform", |
| "unconstrained": false, |
| "use_ema": true, |
| "use_inpainting": false, |
| "use_recon_guidance": false, |
| "weight_decay": 0.0 |
| } |
