From f4f5a96b9f12def2e73e07dfc80a386447ecb596 Mon Sep 17 00:00:00 2001 From: nihuini Date: Fri, 9 Feb 2018 14:08:06 +0800 Subject: [PATCH] fix default concat dim --- tools/mxnet/mxnet2ncnn.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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")