| 2 | |
| 3 | |
| 4 | def get_motion_model(name, ckpt_path): |
| 5 | if name == 'kit_ml': |
| 6 | model = build_submodule(dict( |
| 7 | type='T2MMotionEncoder', |
| 8 | input_size=251, |
| 9 | movement_hidden_size=512, |
| 10 | movement_latent_size=512, |
| 11 | motion_hidden_size=1024, |
| 12 | motion_latent_size=512, |
| 13 | )) |
| 14 | elif name == 'kit_ttc' and 'human' not in ckpt_path: |
| 15 | model = build_submodule(dict( |
| 16 | type='ACTOREncoder', |
| 17 | max_seq_len=196, |
| 18 | input_feats=247, |
| 19 | latent_dim=512, |
| 20 | num_layers=4, |
| 21 | ff_size=2048, |
| 22 | output_var=True |
| 23 | )) |
| 24 | elif name == 'kit_ttc': |
| 25 | model = build_submodule(dict( |
| 26 | type='ACTOREncoder', |
| 27 | max_seq_len=196, |
| 28 | input_feats=133, |
| 29 | latent_dim=512, |
| 30 | num_layers=4, |
| 31 | ff_size=2048, |
| 32 | output_var=True |
| 33 | )) |
| 34 | else: |
| 35 | model = build_submodule(dict( |
| 36 | type='T2MMotionEncoder', |
| 37 | input_size=263, |
| 38 | movement_hidden_size=512, |
| 39 | movement_latent_size=512, |
| 40 | motion_hidden_size=1024, |
| 41 | motion_latent_size=512, |
| 42 | )) |
| 43 | model.load_pretrained(ckpt_path) |
| 44 | return model |
| 45 | |
| 46 | def get_text_model(name, ckpt_path): |
| 47 | if name == 'kit_ml': |