(opt, dim_pose)
| 16 | |
| 17 | |
| 18 | def build_models(opt, dim_pose): |
| 19 | encoder = MotionTransformer( |
| 20 | input_feats=dim_pose, |
| 21 | num_frames=opt.max_motion_length, |
| 22 | num_layers=opt.num_layers, |
| 23 | latent_dim=opt.latent_dim, |
| 24 | no_clip=opt.no_clip, |
| 25 | no_eff=opt.no_eff) |
| 26 | return encoder |
| 27 | |
| 28 | |
| 29 | if __name__ == '__main__': |
no test coverage detected