Method __init__

preprocessing/encoders.py:56–63

(self,
        vae_name    = 'REPA-E/e2e-invae',  # Name of the VAE to use.
        batch_size  = 8,                    # Batch size to use when running the VAE.
    )

@persistence.persistent_class
class InvaeEncoder(Encoder):
    def __init__(self,
        vae_name    = 'REPA-E/e2e-invae',  # Name of the VAE to use.
        batch_size  = 8,                    # Batch size to use when running the VAE.
    ):
        super().__init__()
        self.vae_name = vae_name
        self.batch_size = int(batch_size)
        self._vae = None

    def init(self, device): # force lazy init to happen now
        super().init(device)