Added an option to return a latent embedding to the ConvAutoencoder class + unit test for this case
Compare changes
@@ -12,9 +12,22 @@ from torch import nn
@@ -12,9 +12,22 @@ from torch import nn
@@ -45,5 +58,9 @@ class ConvAutoencoder(nn.Module):
@@ -45,5 +58,9 @@ class ConvAutoencoder(nn.Module):