(self, vae_model, nfuture)
| 124 | class WrappedVAEDecode(torch.nn.Module): |
| 125 | |
| 126 | def __init__(self, vae_model, nfuture): |
| 127 | super().__init__() |
| 128 | self.vae = vae_model |
| 129 | self.nfuture = nfuture |
| 130 | |
| 131 | def forward(self, motion_latent: torch.Tensor, |
| 132 | history_motion: torch.Tensor): |