diff --git a/layers.py b/layers.py
index 55aaab10..e0ee01d4 100644
--- a/layers.py
+++ b/layers.py
@@ -314,10 +314,10 @@ def forward(self, x, y):
     else:
       if self.norm_style == 'bn':
         out = F.batch_norm(x, self.stored_mean, self.stored_var, None, None,
-                          self.training, 0.1, self.eps)
+                          self.training, self.momentum, self.eps)
       elif self.norm_style == 'in':
         out = F.instance_norm(x, self.stored_mean, self.stored_var, None, None,
-                          self.training, 0.1, self.eps)
+                          self.training, self.momentum, self.eps)
       elif self.norm_style == 'gn':
         out = groupnorm(x, self.normstyle)
       elif self.norm_style == 'nonorm':
@@ -456,4 +456,4 @@ def forward(self, x):
         
     return h + self.shortcut(x)
     
-# dogball
\ No newline at end of file
+# dogball