fix(batchnorm): init

marcpinet · Dec 9, 2024 · 2b2e04b · 2b2e04b
1 parent 1ae3158
commit 2b2e04b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neuralnetlib/layers.py b/neuralnetlib/layers.py
@@ -1222,7 +1222,7 @@ def __str__(self) -> str:
         return f'BatchNormalization(momentum={self.momentum}, epsilon={self.epsilon})'
 
     def forward_pass(self, input_data: np.ndarray, training: bool = True) -> np.ndarray:
-        if self.gamma is None:
+        if self.gamma is None or self.running_mean is None or self.running_var is None:
             self.initialize_weights(input_data.shape[1:])
 
         input_data = np.clip(input_data, -10, 10)