Function build_models

text2motion/tools/evaluation.py:19–27

(opt, dim_pose)

def build_models(opt, dim_pose):
    encoder = MotionTransformer(
        input_feats=dim_pose,
        num_frames=opt.max_motion_length,
        num_layers=opt.num_layers,
        latent_dim=opt.latent_dim,
        no_clip=opt.no_clip,
        no_eff=opt.no_eff)
    return encoder


torch.multiprocessing.set_sharing_strategy('file_system')