Fix bug for convolution on x86

Tencent · Sep 18, 2017 · c641db8 · c641db8
1 parent f8dc80f
commit c641db8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/layer/x86/convolution_x86.cpp b/src/layer/x86/convolution_x86.cpp
@@ -26,7 +26,7 @@ int Convolution_x86::forward(const Mat& bottom_blob, Mat& top_blob) const
     // convolv with NxN kernel
     // value = value + bias
 
-    if (kernel_size > 5 || dilation != 1)
+    if (kernel_size > 5 || stride > 5 || dilation != 1)
     {
         return Convolution::forward(bottom_blob, top_blob);
     }