From 5ac2de89636ab39bd8dc618f2d79508d212543a7 Mon Sep 17 00:00:00 2001 From: peng Date: Fri, 29 Dec 2017 15:54:42 +0800 Subject: [PATCH] fix shufflechannel --- src/layer/shufflechannel.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/layer/shufflechannel.cpp b/src/layer/shufflechannel.cpp index f478d1858..47fea2426 100644 --- a/src/layer/shufflechannel.cpp +++ b/src/layer/shufflechannel.cpp @@ -41,8 +41,8 @@ int ShuffleChannel::forward(const Mat &bottom_blob, Mat &top_blob) const size_t feature_sz = w * h * sizeof(float); for (int i = 0; i != group; ++i) { for (int j = 0; j != chs_per_group; ++j) { - dst_q = chs_per_group * i + j; - src_q = chs_per_group * j + i; + src_q = chs_per_group * i + j; + dst_q = group * j + i; memcpy(top_blob.channel(dst_q), bottom_blob.channel(src_q), feature_sz); }