diff --git a/scikinC/layers/BatchNormalization.py b/scikinC/layers/BatchNormalization.py index e76621b..1d4fd32 100644 --- a/scikinC/layers/BatchNormalization.py +++ b/scikinC/layers/BatchNormalization.py @@ -17,7 +17,7 @@ def definition(self): gamma = self.layer.gamma.numpy() beta = self.layer.beta.numpy() mean = self.layer.moving_mean.numpy() - var = self.layer.moving_var.numpy() + var = self.layer.moving_variance.numpy() epsilon = self.layer.epsilon n_inputs = len(gamma)