Browse Source

fix default concat dim

tags/20180314
nihuini 8 years ago
parent
commit
f4f5a96b9f
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      tools/mxnet/mxnet2ncnn.cpp

+ 1
- 1
tools/mxnet/mxnet2ncnn.cpp View File

@@ -1111,7 +1111,7 @@ int main(int argc, char** argv)
}
else if (n.op == "Concat")
{
int dim = n.attr("dim");
int dim = n.has_attr("dim") ? n.attr("dim") : 1;
fprintf(pp, " 0=%d", dim-1);
}
else if (n.op == "Convolution")


Loading…
Cancel
Save