checkpoints/diffusion/c1_pose/args.json

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