models/args.json

{
    "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
}