Browse Source

fix the caffe2ncnn that "DepthwiseConvolution" loss group num param (#540)

* fix the caffe2ncnn  bug that "DepthwiseConvolution" loss group num param
tags/20180830
BUG1989 nihui 7 years ago
parent
commit
d5df2301cb
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      tools/caffe/caffe2ncnn.cpp

+ 1
- 1
tools/caffe/caffe2ncnn.cpp View File

@@ -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();
}


Loading…
Cancel
Save