Skip to content
Snippets Groups Projects

Modified the data_folder to be more generic

Merged Guillaume HEUSCH requested to merge generic_loader into master
Files
8
@@ -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
Loading