fix filter sizes in bottleneck block (#42)

raghakot · Jul 5, 2017 · 5e9bcca · 5e9bcca
1 parent 9a2685a
commit 5e9bcca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/resnet.py b/resnet.py
@@ -147,7 +147,7 @@ def f(input):
                               kernel_initializer="he_normal",
                               kernel_regularizer=l2(1e-4))(input)
         else:
-            conv_1_1 = _bn_relu_conv(filters=filters, kernel_size=(3, 3),
+            conv_1_1 = _bn_relu_conv(filters=filters, kernel_size=(1, 1),
                                      strides=init_strides)(input)
 
         conv_3_3 = _bn_relu_conv(filters=filters, kernel_size=(3, 3))(conv_1_1)