Added an option to return a latent embedding to the ConvAutoencoder class + unit test for this case
Compare changes
Files
2@@ -12,9 +12,22 @@ from torch import nn
@@ -45,5 +58,9 @@ class ConvAutoencoder(nn.Module):
Following our today's discussion, here is the update to the autoencoder class allowing to avoid prediction_function
in the configs. Please merge if fine with you. Thanks!