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

fix: dup load method

Browse Source
This commit is contained in:
Lenoctambule
2026-04-05 08:41:17 +02:00
parent 82d61dd10f
commit 3440de851a
2 changed files with 2 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
layers.py
View File

@@ -59,12 +59,12 @@ class SampleLayer:
self.input = v
self.mean = self.mean_nn.forward(v)
self.std = self.std_nn.forward(v)
self.eps = np.random.normal(0, 1)
self.eps = np.random.normal(0, 1, self.mean.shape)
return self.eps * self.std + self.mean
def backprop(self, error: np.ndarray) -> np.ndarray:
mu_error = self.mean_nn.backprop(error)
std_error = self.std_nn.backprop(self.eps * error)
std_error = self.std_nn.backprop(error * self.eps * self.std * 0.5)
return mu_error + std_error