complex_shift_autoencoder.py [48:67]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        n_rotations = 0,
        n_x_translations = 0,
        n_y_translations = 0,
        scaling_factors = (1, )
    ):

        self.z_dim = z_dim

        self.seed = seed
        self.set_seed()

        self.data = data
        self.device = device

        self.encoder = getattr(models, encoder_type + "Encoder")(
            self.data.n_pixels, self.data.n_channels, z_dim
        ).to(self.device)
        self.decoder = getattr(models, decoder_type + "Decoder")(
        self.data.n_pixels, self.data.n_channels, z_dim
        ).to(self.device)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



weakly_complex_shift_autoencoder.py [50:69]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        n_rotations = 0,
        n_x_translations = 0,
        n_y_translations = 0,
        scaling_factors = (1, )
    ):

        self.z_dim = z_dim

        self.seed = seed
        self.set_seed()

        self.data = data
        self.device = device

        self.encoder = getattr(models, encoder_type + "Encoder")(
            self.data.n_pixels, self.data.n_channels, z_dim
        ).to(self.device)
        self.decoder = getattr(models, decoder_type + "Decoder")(
        self.data.n_pixels, self.data.n_channels, z_dim
        ).to(self.device)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -