(self, input_size, movement_hidden_size, movement_latent_size,
motion_hidden_size, motion_latent_size)
| 71 | class T2MMotionEncoder(nn.Module): |
| 72 | |
| 73 | def __init__(self, input_size, movement_hidden_size, movement_latent_size, |
| 74 | motion_hidden_size, motion_latent_size): |
| 75 | super().__init__() |
| 76 | self.movement_encoder = MovementConvEncoder( |
| 77 | input_size=input_size - 4, |
| 78 | hidden_size=movement_hidden_size, |
| 79 | output_size=movement_latent_size) |
| 80 | self.motion_encoder = MotionEncoderBiGRUCo( |
| 81 | input_size=movement_latent_size, |
| 82 | hidden_size=motion_hidden_size, |
| 83 | output_size=motion_latent_size) |
| 84 | |
| 85 | def load_pretrained(self, ckpt_path): |
| 86 | checkpoint = torch.load(ckpt_path, map_location='cpu') |
nothing calls this directly
no test coverage detected