Spaces:
Runtime error
Runtime error
File size: 1,627 Bytes
882f6e2 |
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 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
{
"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
}
|