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

fix: missing args in np.clip

Browse Source
This commit is contained in:
Lenoctambule
2026-04-08 14:49:56 +02:00
parent 510ad8720c
commit e9c79f463f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
layers.py
View File

@@ -61,7 +61,7 @@ class SampleLayer:
def forward(self, v: np.ndarray) -> np.ndarray: def forward(self, v: np.ndarray) -> np.ndarray:
self.input = v self.input = v
self.mean = self.mean_nn.forward(v) self.mean = self.mean_nn.forward(v)
self.logvar = np.clip(self.std_nn.forward(v)) self.logvar = np.clip(self.std_nn.forward(v), -10, 10)
self.std = np.exp(0.5 * self.logvar) self.std = np.exp(0.5 * self.logvar)
self.eps = np.random.normal(0, 1, self.mean.shape) self.eps = np.random.normal(0, 1, self.mean.shape)
return 0.5 * self.eps * self.std + self.mean return 0.5 * self.eps * self.std + self.mean