NamYeongCho commited on
Commit
951ecba
·
verified ·
1 Parent(s): 0fe9602

Upload args.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. args.json +65 -0
args.json ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "action_emb": null,
3
+ "arch": "trans_enc",
4
+ "attention_resolutions": "16,8",
5
+ "batch_size": 64,
6
+ "channel_mult": "",
7
+ "class_cond": false,
8
+ "cond_mask_prob": 0.1,
9
+ "cond_mode": "text",
10
+ "data_dir": "",
11
+ "data_rep": "hml_vec",
12
+ "dataset": "humanml",
13
+ "device": 5,
14
+ "diffusion_steps": 1000,
15
+ "dropout": 0.0,
16
+ "ema_rate": "0.9999",
17
+ "emb_trans_dec": false,
18
+ "eval_batch_size": 32,
19
+ "eval_debug": false,
20
+ "eval_num_samples": -1,
21
+ "eval_rep_times": 3,
22
+ "eval_split": "test",
23
+ "fp16_scale_growth": 0.001,
24
+ "image_size": 64,
25
+ "lambda_fc": 0.0,
26
+ "lambda_loc": 1.0,
27
+ "lambda_orient": 1.0,
28
+ "lambda_pose": 1.0,
29
+ "lambda_rcxyz": 0.0,
30
+ "lambda_vel": 0.0,
31
+ "latent_dim": 512,
32
+ "layers": 8,
33
+ "learn_sigma": false,
34
+ "legacy": false,
35
+ "log_interval": 1000,
36
+ "lr": 0.0001,
37
+ "lr_anneal_steps": 0,
38
+ "microbatch": -1,
39
+ "noise_schedule": "cosine",
40
+ "num_channels": 128,
41
+ "num_epochs": 100000,
42
+ "num_head_channels": -1,
43
+ "num_heads": 4,
44
+ "num_heads_upsample": -1,
45
+ "num_res_blocks": 2,
46
+ "overwrite": false,
47
+ "predict_xstart": true,
48
+ "resblock_updown": false,
49
+ "rescale_learned_sigmas": false,
50
+ "rescale_timesteps": false,
51
+ "resume_checkpoint": "",
52
+ "save_dir": "save/humanml_trans_enc_512",
53
+ "save_interval": 25000,
54
+ "scale_beta": 1.0,
55
+ "schedule_sampler": "uniform",
56
+ "sigma_small": true,
57
+ "timestep_respacing": "",
58
+ "train_platform_type": "ClearmlPlatform",
59
+ "use_checkpoint": false,
60
+ "use_fp16": false,
61
+ "use_kl": false,
62
+ "use_new_attention_order": false,
63
+ "use_scale_shift_norm": true,
64
+ "weight_decay": 0.0
65
+ }