bug fix with conv input params

ragavvenkatesan · Mar 13, 2017 · 8327d05 · 8327d05
1 parent ea10159
commit 8327d05
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/yann/layers/conv_pool.py b/yann/layers/conv_pool.py
@@ -87,6 +87,10 @@ def __init__ ( self,
             if batch_norm is True:
                 if input_params [2] is None:
                     create_bn = True
+        else:
+            create_w = True
+            create_b = True
+            create_bn = True
 
         mini_batch_size  = input_shape[0]
         channels   = input_shape[1]
@@ -404,7 +408,11 @@ def __init__ ( self,
             if batch_norm is True:
                 if input_params [2] is None:
                     create_bn = True
-
+        else:
+            create_w = True
+            create_b = True
+            create_bn = True
+
         mini_batch_size  = input_shape[0]
         channels   = input_shape[1]
         width      = input_shape[3]