SemGeoMo/args.json at main · 4DVLab/SemGeoMo · GitHub

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
{
    "arch": "trans_enc",
    "batch_size": 16,
    "bfgs_interval": 1,
    "bfgs_times_first": 5,
    "bfgs_times_last": 10,
    "concat_trans_emb": false,
    "cond_mask_prob": 0.1,
    "control_joint": ["left_wrist","right_wrist"],
    "cropping_sampler": false,
    "cuda": true,
    "data_dir": "",
    "dataset": "omomo",
    "device": 0,
    "diffusion_steps": 1000,
    "emb_trans_dec": false,
    "eval_batch_size": 32,
    "eval_during_training": false,
    "eval_num_samples": 1000,
    "eval_rep_times": 3,
    "eval_split": "test",
    "filter_noise": true,
    "inpainting_mask": "global_joint",
    "lambda_fc": 0.0,
    "lambda_rcxyz": 0.0,
    "lambda_vel": 0.0,
    "latent_dim": 512,
    "layers": 8,
    "log_interval": 1000,
    "lr": 1e-05,
    "lr_anneal_steps": 0,
    "mask_ratio": 1.0,
    "multi_person": false,
    "noise_schedule": "cosine",
    "num_frames": 100,
    "num_steps": 1000000,
    "overwrite": false,
    "pretrained_path": "/storage/group/4dvlab/code_share/uni_regen/pretrain/OMOMO_fps15/model000400000.pt",
    "resume_checkpoint": "",
    "save_dir": " ",
    "save_interval": 10000,
    "seed": 10,
    "short_db": false,
    "sigma_small": true,
    "train_platform_type": "NoPlatform",
    "trans_emb": false,
    "use_posterior": true,
    "use_tta": false,
    "weight_decay": 0.0
}