def init_decoder(self): decoder = Decoder( self.hparams.hidden_dim, self.hparams.latent_dim, self.hparams.input_width, self.hparams.input_height, self.hparams.input_channels, ) return decoder
def init_decoder(self, hidden_dim, latent_dim): c, h, w = self.img_dim decoder = Decoder(hidden_dim, latent_dim, w, h, c) return decoder
def init_decoder(self, hidden_dim, latent_dim): # c, h, w = self.img_dim decoder = Decoder(hidden_dim, latent_dim, self.hparams.input_width, self.hparams.input_height, self.hparams.input_channels) return decoder
def init_decoder(self, hidden_dim, latent_dim, input_width, input_height): decoder = Decoder(hidden_dim, latent_dim, input_width, input_height) return decoder