motius-condmdi-humanml3d / condmdi_config.json
ZeyuLing's picture
Add files using upload-large-folder tool
5360acc verified
Raw History Blame Contribute Delete
2.15 kB
{
"model_type": "condmdi",
"library_name": "motius",
"tasks": [
"text-to-motion",
"motion-control",
"motion-in-betweening"
],
"native_motion_representation": "HumanML3D-263 absolute-root",
"guidance_param": 2.5,
"respacing": "",
"source_revision": "model_avg",
"config": {
"dataset": "humanml",
"arch": "unet",
"latent_dim": 512,
"ff_size": 1024,
"layers": 8,
"dim_mults": [
2,
2,
2,
2
],
"cond_mask_prob": 0.1,
"emb_trans_dec": false,
"noise_schedule": "cosine",
"diffusion_steps": 1000,
"sigma_small": true,
"predict_xstart": true,
"clip_range": 6.0,
"unet_adagn": true,
"unet_zero": true,
"out_mult": false,
"xz_only": false,
"keyframe_conditioned": true,
"keyframe_selection_scheme": "random_frames",
"zero_keyframe_loss": false,
"use_fp16": true,
"abs_3d": true,
"lambda_rcxyz": 0.0,
"lambda_vel": 0.0,
"lambda_fc": 0.0,
"use_random_proj": false,
"traj_only": false,
"apply_zero_mask": false,
"traj_extra_weight": 1.0,
"time_weighted_loss": false,
"train_x0_as_eps": false,
"adam_beta2": 0.999,
"augment_type": "none",
"avg_model_beta": 0.9999,
"batch_size": 64,
"cuda": true,
"data_dir": "",
"device": 0,
"drop_redundant": false,
"eval_batch_size": 32,
"eval_during_training": false,
"eval_mode": "wo_mm",
"eval_num_samples": 1000,
"eval_rep_times": 3,
"eval_split": "test",
"eval_use_avg": true,
"full_traj_inpaint": false,
"grad_clip": 1.0,
"guidance_param": 2.5,
"impute_until": null,
"log_interval": 1000,
"lr": 0.0001,
"lr_anneal_steps": 0,
"model_path": "",
"num_frames": 224,
"num_steps": 3000000,
"overwrite": false,
"random_proj_scale": 10.0,
"resume_checkpoint": "",
"save_dir": "save/xpnzs9fw",
"save_interval": 50000,
"seed": 10,
"skip_first": null,
"std_scale_shift": [
1.0,
0.0
],
"train_platform_type": "NoPlatform",
"unconstrained": false,
"use_ddim": false,
"weight_decay": 0.01
}
}