(opt)
| 26 | |
| 27 | |
| 28 | def build_models(opt): |
| 29 | encoder = MotionTransformer( |
| 30 | input_feats=opt.dim_pose, |
| 31 | num_frames=opt.max_motion_length, |
| 32 | num_layers=opt.num_layers, |
| 33 | latent_dim=opt.latent_dim, |
| 34 | no_clip=opt.no_clip, |
| 35 | no_eff=opt.no_eff) |
| 36 | return encoder |
| 37 | |
| 38 | |
| 39 | if __name__ == '__main__': |
no test coverage detected