Fix AutoEncoder calling pre/postprocessing during training

luigibonati · May 23, 2023 · 6b21e3e · 6b21e3e
1 parent 98a5dcf
commit 6b21e3e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mlcolvar/cvs/unsupervised/autoencoder.py b/mlcolvar/cvs/unsupervised/autoencoder.py
@@ -80,7 +80,7 @@ def forward_cv(self, x: torch.Tensor) -> (torch.Tensor):
         return x
 
     def encode_decode(self, x: torch.Tensor) -> (torch.Tensor):
-        x = self.forward(x)
+        x = self.forward_cv(x)
         x = self.decoder(x)
         if self.norm_in is not None:
             x = self.norm_in.inverse(x)
@@ -161,4 +161,4 @@ def test_autoencodercv():
     trainer.fit( model, datamodule )
 
 if __name__ == "__main__":
-    test_autoencodercv() 
+    test_autoencodercv()