Method encode

diffplanner/models/architectures/vae_architecture.py:48–50

(self, motion)

        return latent_code

    def encode(self, motion):
        mu, logvar = self.encoder(motion)
        return mu

    def forward(self, **kwargs):
        motion, motion_mask = kwargs['motion'], kwargs['motion_mask']