Upload 5 files

models/args.json

@@ -0,0 +1,65 @@
+{
+    "action_emb": null,
+    "arch": "trans_enc",
+    "attention_resolutions": "16,8",
+    "batch_size": 64,
+    "channel_mult": "",
+    "class_cond": false,
+    "cond_mask_prob": 0.1,
+    "cond_mode": "text",
+    "data_dir": "",
+    "data_rep": "hml_vec",
+    "dataset": "humanml",
+    "device": 5,
+    "diffusion_steps": 1000,
+    "dropout": 0.0,
+    "ema_rate": "0.9999",
+    "emb_trans_dec": false,
+    "eval_batch_size": 32,
+    "eval_debug": false,
+    "eval_num_samples": -1,
+    "eval_rep_times": 3,
+    "eval_split": "test",
+    "fp16_scale_growth": 0.001,
+    "image_size": 64,
+    "lambda_fc": 0.0,
+    "lambda_loc": 1.0,
+    "lambda_orient": 1.0,
+    "lambda_pose": 1.0,
+    "lambda_rcxyz": 0.0,
+    "lambda_vel": 0.0,
+    "latent_dim": 512,
+    "layers": 8,
+    "learn_sigma": false,
+    "legacy": false,
+    "log_interval": 1000,
+    "lr": 0.0001,
+    "lr_anneal_steps": 0,
+    "microbatch": -1,
+    "noise_schedule": "cosine",
+    "num_channels": 128,
+    "num_epochs": 100000,
+    "num_head_channels": -1,
+    "num_heads": 4,
+    "num_heads_upsample": -1,
+    "num_res_blocks": 2,
+    "overwrite": false,
+    "predict_xstart": true,
+    "resblock_updown": false,
+    "rescale_learned_sigmas": false,
+    "rescale_timesteps": false,
+    "resume_checkpoint": "",
+    "save_dir": "save/humanml_trans_enc_512",
+    "save_interval": 25000,
+    "scale_beta": 1.0,
+    "schedule_sampler": "uniform",
+    "sigma_small": true,
+    "timestep_respacing": "",
+    "train_platform_type": "ClearmlPlatform",
+    "use_checkpoint": false,
+    "use_fp16": false,
+    "use_kl": false,
+    "use_new_attention_order": false,
+    "use_scale_shift_norm": true,
+    "weight_decay": 0.0
+}

models/model000200000.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d86e1eb3b9fa10456c81e97701dc77e1b6b3cd4909bb6d6651f9460e8db29373
+size 81806834

models/model000475000.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ce43c1d4232fc2b897999d7996c2d2b889791d99585d87e1289af773ab56ab93
+size 81806834

models/opt000200000.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:163a7db0749667d07550d1ea6a13ab0b01fa06003c103a370fb944fea1da2578
+size 143109219

models/opt000475000.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3d01006a0cb01b358bae65ff451e57ed58ab715815fdce5f06c6f498c1846941
+size 143109219