Method encode

preprocessing/encoders.py:79–83

(self, x)

        return self._vae.encode(x).sample()

    def encode(self, x): # raw pixels => raw latents
        self.init(x.device)
        x = x.to(torch.float32) / 127.5 - 1
        x = torch.cat([self._run_vae_encoder(batch) for batch in x.split(self.batch_size)])
        return x

#----------------------------------------------------------------------------