lenoctambule/autoencoder
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: return encoding after sample layer

Browse Source
This commit is contained in:
Lenoctambule
2026-04-13 23:07:11 +02:00
parent 82acc6b4f3
commit e6c1229a7e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/easyvae/autoencoder.py
View File

@@ -232,7 +232,7 @@ class VariationalAutoencoder(AAutoencoder):
code = self.encoder.forward(v) code = self.encoder.forward(v)
sample = self.sampler.forward(code) sample = self.sampler.forward(code)
out = self.decoder.forward(sample) out = self.decoder.forward(sample)
return out, code return out, sample
def encode(self, v: np.ndarray) -> np.ndarray: def encode(self, v: np.ndarray) -> np.ndarray:
return self.sampler.forward( return self.sampler.forward(