diff --git a/tools/mxnet/mxnet2ncnn.cpp b/tools/mxnet/mxnet2ncnn.cpp index c651228d6..134ac79b7 100644 --- a/tools/mxnet/mxnet2ncnn.cpp +++ b/tools/mxnet/mxnet2ncnn.cpp @@ -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")