fix the caffe2ncnn that "DepthwiseConvolution" loss group num param (T…

…encent#540) * fix the caffe2ncnn bug that "DepthwiseConvolution" loss group num param
JullyZ · Aug 23, 2018 · d5df230 · d5df230
1 parent 9e49144
commit d5df230
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/caffe/caffe2ncnn.cpp b/tools/caffe/caffe2ncnn.cpp
@@ -638,7 +638,7 @@ int main(int argc, char** argv)
             fprintf(pp, " 6=%d", weight_blob.data_size());
 
             int num_group = 1;
-            if (layer.type() == "ConvolutionDepthwise")
+            if (layer.type() == "ConvolutionDepthwise" || layer.type() == "DepthwiseConvolution")
             {
                 num_group = convolution_param.num_output();
             }