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