Commit 7ef52d8b authored by Olegs NIKISINS's avatar Olegs NIKISINS
Browse files

Fixed a bug in the ConvAutoencoder

parent 43fadd2a
Pipeline #26440 passed with stage
in 7 minutes and 29 seconds
......@@ -57,10 +57,11 @@ class ConvAutoencoder(nn.Module):
The forward method.
"""
x = self.encoder(x)
x = self.decoder(x)
if self.return_latent_embedding:
return self.encoder(x)
return x
x = self.decoder(x)
return x
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment