From 3440de851ad405bf1668a8d6ee92d6f4ead0d0cb Mon Sep 17 00:00:00 2001
From: Lenoctambule <106790775+lenoctambule@users.noreply.github.com>
Date: Sun, 5 Apr 2026 08:41:17 +0200
Subject: [PATCH] fix: dup load method

---
 autoencoder.py | 5 -----
 layers.py      | 4 ++--
 2 files changed, 2 insertions(+), 7 deletions(-)

diff --git a/autoencoder.py b/autoencoder.py
index 910efde..dbb4f8d 100644
--- a/autoencoder.py
+++ b/autoencoder.py
@@ -137,11 +137,6 @@ class VariationalAutoencoder(AAutoencoder):
             loss += np.sum(np.abs(x - self.forward(x)[0])) / len(x)
         return loss / len(data_set)
 
-    def load(path: str) -> 'ClassicalAutoencoder':
-        path = path.removesuffix('.npy') + '.npy'
-        data = np.load(path, allow_pickle=True)
-        return data.item()
-
     def train(self, v: np.ndarray) -> float:
         out, _ = self.forward(v)
         error = out - v
diff --git a/layers.py b/layers.py
index 7591d3f..e2a3b08 100644
--- a/layers.py
+++ b/layers.py
@@ -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