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