{ "add_frame_cond": 1.0, "arch": "trans_enc", "batch_size": 32, "clip_body": false, "clip_use_delta": false, "clip_use_vae": false, "cond_mask_prob": 0.1, "cuda": true, "data_format": "pose", "data_root": "./dataset/PXB184/", "dataset": "social", "device": 0, "diffusion_steps": 10, "emb_trans_dec": false, "eval_batch_size": 32, "eval_during_training": false, "eval_num_samples": 1000, "eval_rep_times": 3, "eval_split": "val", "filter": false, "heads": 8, "lambda_fc": 0.0, "lambda_hands": 0.0, "lambda_lips": 0.0, "lambda_rcxyz": 0.0, "lambda_vel": 2.0, "lambda_xyz": 0.0, "lambda_xyz_vel": 0.0, "latent_dim": 512, "layers": 6, "log_interval": 1000, "lr": 0.0001, "lr_anneal_steps": 0, "max_seq_length": 600, "no_split": false, "noise_schedule": "cosine", "not_rotary": false, "num_frames": 60, "num_steps": 800000, "overwrite": false, "partial": false, "resume_checkpoint": "", "save_dir": "checkpoints/diffusion/c1_pose/", "save_interval": 5000, "seed": 10, "sigma_small": true, "simplify_audio": false, "split_net": false, "timestep_respacing": "", "train_platform_type": "NoPlatform", "unconstrained": false, "use_clip": false, "use_cm": true, "use_full_dataset": false, "use_kp": false, "use_mask": true, "use_mdm": false, "use_nort": false, "use_nort_mdm": false, "use_pose_pos": false, "use_resnet": true, "use_vae": null, "weight_decay": 0.0, "z_norm": true }