| { |
| "activation": "gelu", |
| "adam_beta2": 0.999, |
| "arch": "trans_enc", |
| "avg_model_beta": 0.9999, |
| "batch_size": 32, |
| "cache_embed_text": true, |
| "cond_mask_prob": 0.1, |
| "cond_mode": "action", |
| "data_rep": "smal_vec", |
| "dataset": "action_motion", |
| "device": "cuda:3", |
| "diffusion_steps": 50, |
| "dropout": 0.1, |
| "eval_batch_size": 32, |
| "eval_during_training": false, |
| "eval_num_samples": 1000, |
| "eval_rep_times": 3, |
| "eval_split": "test", |
| "ff_size": 1024, |
| "gen_during_training": false, |
| "gen_guidance_param": 2.5, |
| "gen_num_repetitions": 2, |
| "gen_num_samples": 3, |
| "lambda_fc": 0.0, |
| "lambda_rcxyz": 0.0, |
| "lambda_target_loc": 0.0, |
| "lambda_vel": 0.0, |
| "latent_dim": 512, |
| "log_interval": 500, |
| "lr": 0.0001, |
| "lr_anneal_steps": 0, |
| "mask_frames": true, |
| "nfeats": 1, |
| "njoints": 251, |
| "noise_schedule": "cosine", |
| "num_actions": 15, |
| "num_frames": 60, |
| "num_heads": 4, |
| "num_layers": 8, |
| "num_steps": 360000, |
| "overwrite": false, |
| "pos_embed_max_len": 5000, |
| "resume_checkpoint": "", |
| "save_dir": "save/baseline_steps_50_a2m_3.10", |
| "save_interval": 20000, |
| "seed": 10, |
| "sigma_small": true, |
| "text_encoder_type": "clip", |
| "uncond": false, |
| "use_ema": true, |
| "weight_decay": 0.0 |
| } |
