(self, ckpt_path)
| 83 | output_size=motion_latent_size) |
| 84 | |
| 85 | def load_pretrained(self, ckpt_path): |
| 86 | checkpoint = torch.load(ckpt_path, map_location='cpu') |
| 87 | self.movement_encoder.load_state_dict(checkpoint['movement_encoder']) |
| 88 | self.motion_encoder.load_state_dict(checkpoint['motion_encoder']) |
| 89 | |
| 90 | def forward(self, motion, motion_length, motion_mask): |
| 91 | motion = motion.detach().float() |
nothing calls this directly
no outgoing calls
no test coverage detected