Merge pull request #27 from JDAI-CV/fix_128_s1_internal

Fix 128_internal_s1 called when tw % 2 == 1
JDAI-CV · May 22, 2019 · 9e40eda · 9e40eda
2 parents 7f710f0 + 281479b
commit 9e40eda
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dabnn/bconv.h b/dabnn/bconv.h
@@ -833,7 +833,7 @@ inline void bnn::bconv_3x3(const Mat &bottom_blob, const Mat &weight,
         bconv_3x3_64(bottom_blob, weight, top_blob, stride);
     } else if (bottom_blob.c == 2 && top_blob.c == 128) {
         top_blob.fill<float>(0.f);
-        if (stride == 1) {
+        if (stride == 1 && top_blob.w % 2 == 0) {
             bconv_3x3_128_internal_s1(
                 static_cast<uint64_t *>(bottom_blob.data), bottom_blob.w,
                 static_cast<uint64_t *>(weight.data),