fix: return encoding after sample layer

2026-04-13 23:07:11 +02:00
parent 82acc6b4f3
commit e6c1229a7e
1 changed files with 1 additions and 1 deletions
--- a/src/easyvae/autoencoder.py
+++ b/src/easyvae/autoencoder.py
@@ -232,7 +232,7 @@ class VariationalAutoencoder(AAutoencoder):
        code = self.encoder.forward(v)
        sample = self.sampler.forward(code)
        out = self.decoder.forward(sample)
-        return out, code
+        return out, sample

    def encode(self, v: np.ndarray) -> np.ndarray:
        return self.sampler.forward(