config.yml

#Don't change "config" fields plz
model:
    remodiffuse:
        config: 
            human_ml3d: "configs/remodiffuse/remodiffuse_t2m.py"
            kit_ml: "configs/remodiffuse/remodiffuse_kit.py"
        ckpt: 
            human_ml3d: "ckpts/remodiffuse_t2m.pth"
            kit_ml: "ckpts/remodiffuse_kit.pth"

    motiondiffuse:
        config: 
            human_ml3d: "configs/motiondiffuse/motiondiffuse_t2m.py"
            kit_ml: "configs/motiondiffuse/motiondiffuse_kit.py"
        ckpt: 
            human_ml3d: "ckpts/motiondiffuse_t2m.pth"
            kit_ml: "ckpts/motiondiffuse_kit.pth"

    mdm:
        config: 
            human_ml3d: "configs/mdm/mdm_t2m_official.py"
            kit_ml: "configs/mdm/mdm_kit_official.py"
        ckpt: 
            human_ml3d: "ckpts/mdm_t2m.pth"
            kit_ml: "ckpts/mdm_kit.pth"


dataset: 
    human_ml3d:
        path: "datasets/human_ml3d/"
        retrieval_db: "retrieval_dbs/t2m_text_train.npz"
        base: "configs/_base_/datasets/human_ml3d_bs128.py"
    kit_ml:
        path: "datasets/kit_ml/"
        retrieval_db: "retrieval_dbs/kit_text_train.npz"
        base: "configs/_base_/datasets/kit_ml_bs128.py"

smpl_models: "smpl_models/"