Variational-Autoencoder-PyTorch icon indicating copy to clipboard operation
Variational-Autoencoder-PyTorch copied to clipboard

Input reshape

Open cutlass90 opened this issue 7 years ago • 0 comments

Why you reshape input before passing to encoder?

def forward(self, x):
      mu, logvar = self.encode(x.view(-1, self.nc, self.ndf, self.ngf))

line 149 self.ndf, self.ngf - are numbers of filters(channels). Am I wrong?

cutlass90 avatar Apr 17 '18 13:04 cutlass90