Spaces:
Paused
Paused
File size: 1,682 Bytes
8a8634d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
{
"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
}
|