guytevet commited on
Commit
c469b52
Β·
1 Parent(s): 5f76523

added dip checkpoints

Browse files
checkpoints/dip/DiP_multi-target_10steps_context20_predict40/args.json ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "adam_beta2": 0.999,
3
+ "arch": "trans_dec",
4
+ "avg_model_beta": 0.9999,
5
+ "batch_size": 64,
6
+ "cond_mask_prob": 0.1,
7
+ "context_len": 20,
8
+ "cuda": true,
9
+ "data_dir": "",
10
+ "dataset": "humanml",
11
+ "device": 0,
12
+ "diffusion_steps": 10,
13
+ "emb_before_mask": false,
14
+ "emb_policy": "concat",
15
+ "emb_trans_dec": false,
16
+ "eval_batch_size": 32,
17
+ "eval_during_training": true,
18
+ "eval_num_samples": 1000,
19
+ "eval_rep_times": 3,
20
+ "eval_split": "test",
21
+ "gen_during_training": true,
22
+ "gen_guidance_param": 2.5,
23
+ "gen_num_repetitions": 1,
24
+ "gen_num_samples": 6,
25
+ "hml_type": null,
26
+ "keyframe_cond_prob": 0.5,
27
+ "keyframe_cond_type": "",
28
+ "lambda_fc": 0.0,
29
+ "lambda_rcxyz": 0.0,
30
+ "lambda_target_loc": 1.0,
31
+ "lambda_vel": 0.0,
32
+ "latent_dim": 512,
33
+ "layers": 8,
34
+ "log_interval": 1000,
35
+ "lr": 0.0001,
36
+ "lr_anneal_steps": 0,
37
+ "mask_frames": true,
38
+ "multi_encoder_type": "single",
39
+ "multi_target_cond": true,
40
+ "noise_schedule": "cosine",
41
+ "num_frames": 60,
42
+ "num_steps": 600000,
43
+ "overwrite": true,
44
+ "pos_embed_max_len": 5000,
45
+ "pred_len": 40,
46
+ "resume_checkpoint": "",
47
+ "sampling_mode": "none",
48
+ "save_dir": "save/MultiJointSearch_EmbPolicy_concat_MultiEncoderType_single_TargetEncLayers_1",
49
+ "save_interval": 50000,
50
+ "seed": 10,
51
+ "sigma_small": true,
52
+ "spatial_condition": null,
53
+ "target_enc_layers": 1,
54
+ "target_joint_names": "DIMP_FULL",
55
+ "text_encoder_type": "bert",
56
+ "train_platform_type": "WandBPlatform",
57
+ "unconstrained": false,
58
+ "use_ema": true,
59
+ "use_inpainting": false,
60
+ "use_recon_guidance": false,
61
+ "weight_decay": 0.0
62
+ }
checkpoints/dip/DiP_multi-target_10steps_context20_predict40/model000300000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3ca68ae54975946b3fe5c6cd64a6eafc439b5fc43c166bc9a98d1c43aeef6618
3
+ size 234251013
checkpoints/dip/DiP_multi-target_10steps_context20_predict40/opt000300000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:425a90e861372e818be663857520fabe639afae5f235f1188d9a1237d0a1514d
3
+ size 213765645
checkpoints/dip/{DiP_10steps_context20_predict40 β†’ DiP_no-target_10steps_context20_predict40}/args.json RENAMED
File without changes
checkpoints/dip/{DiP_10steps_context20_predict40 β†’ DiP_no-target_10steps_context20_predict40}/eval_humanml_PreComHorRerAut_DifSte_10_ConLen_20_PreLen_40_000600343_gscale7.5_wo_mm.log RENAMED
File without changes
checkpoints/dip/{DiP_10steps_context20_predict40 β†’ DiP_no-target_10steps_context20_predict40}/model000200000.pt RENAMED
File without changes
checkpoints/dip/{DiP_10steps_context20_predict40 β†’ DiP_no-target_10steps_context20_predict40}/model000600343.pt RENAMED
File without changes
checkpoints/dip/{DiP_10steps_context20_predict40 β†’ DiP_no-target_10steps_context20_predict40}/opt000200000.pt RENAMED
File without changes
checkpoints/dip/{DiP_10steps_context20_predict40 β†’ DiP_no-target_10steps_context20_predict40}/opt000600343.pt RENAMED
File without changes