Exemplo n.º 1
0
 def init_encoder(self):
     encoder = Encoder(
         self.hparams.hidden_dim,
         self.hparams.latent_dim,
         self.hparams.input_channels,
         self.hparams.input_width,
         self.hparams.input_height,
     )
     return encoder
 def init_encoder(self, hidden_dim, latent_dim, input_width, input_height):
     encoder = Encoder(hidden_dim, latent_dim, input_width, input_height)
     return encoder