megalado commited on
Commit
10dfbcb
·
1 Parent(s): d345528

add args.json required by generate.py

Browse files
Files changed (1) hide show
  1. checkpoints/args.json +51 -0
checkpoints/args.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "adam_beta2": 0.999,
3
+ "arch": "trans_enc",
4
+ "avg_model_beta": 0.9999,
5
+ "batch_size": 64,
6
+ "cond_mask_prob": 0.1,
7
+ "context_len": 0,
8
+ "cuda": true,
9
+ "data_dir": "",
10
+ "dataset": "humanml",
11
+ "device": 0,
12
+ "diffusion_steps": 50,
13
+ "emb_before_mask": false,
14
+ "emb_trans_dec": false,
15
+ "eval_batch_size": 32,
16
+ "eval_during_training": true,
17
+ "eval_num_samples": 1000,
18
+ "eval_rep_times": 3,
19
+ "eval_split": "test",
20
+ "gen_during_training": true,
21
+ "gen_guidance_param": 2.5,
22
+ "gen_num_repetitions": 2,
23
+ "gen_num_samples": 3,
24
+ "keyframe_cond_prob": 0.5,
25
+ "keyframe_cond_type": "",
26
+ "lambda_fc": 0.0,
27
+ "lambda_rcxyz": 0.0,
28
+ "lambda_vel": 0.0,
29
+ "latent_dim": 512,
30
+ "layers": 8,
31
+ "log_interval": 1000,
32
+ "lr": 0.0001,
33
+ "lr_anneal_steps": 0,
34
+ "mask_frames": false,
35
+ "noise_schedule": "cosine",
36
+ "num_frames": 60,
37
+ "num_steps": 600000,
38
+ "overwrite": true,
39
+ "pos_embed_max_len": 5000,
40
+ "pred_len": 0,
41
+ "resume_checkpoint": "",
42
+ "save_dir": "save/BaselinesMain_DiffusionSteps_50",
43
+ "save_interval": 50000,
44
+ "seed": 10,
45
+ "sigma_small": true,
46
+ "text_encoder_type": "clip",
47
+ "train_platform_type": "WandBPlatform",
48
+ "unconstrained": false,
49
+ "use_ema": false,
50
+ "weight_decay": 0.0
51
+ }