File size: 1,331 Bytes
810faa3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 | {
"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
} |