From 281479b2721cfaa7f53e00a59f6fd35cce8d6e5d Mon Sep 17 00:00:00 2001 From: daquexian Date: Wed, 22 May 2019 12:06:19 +0800 Subject: [PATCH] Fix 128_internal_s1 called when tw % 2 == 1 --- dabnn/bconv.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dabnn/bconv.h b/dabnn/bconv.h index 4d1fadc..3dd43c0 100644 --- 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(0.f); - if (stride == 1) { + if (stride == 1 && top_blob.w % 2 == 0) { bconv_3x3_128_internal_s1( static_cast(bottom_blob.data), bottom_blob.w, static_cast(weight.data),