From d06606b0cd2333f26deea1dad97645e151d3dc8c Mon Sep 17 00:00:00 2001 From: mengfu188 <1242733702@qq.com> Date: Mon, 13 Apr 2020 11:28:48 +0800 Subject: [PATCH] Tiny fix mxnet2ncnn tool (#1691) --- tools/mxnet/mxnet2ncnn.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tools/mxnet/mxnet2ncnn.cpp b/tools/mxnet/mxnet2ncnn.cpp index d77315a5e..15e7f7c41 100644 --- a/tools/mxnet/mxnet2ncnn.cpp +++ b/tools/mxnet/mxnet2ncnn.cpp @@ -2387,7 +2387,8 @@ int main(int argc, char** argv) // skip N-dim begin.erase(begin.begin()); end.erase(end.begin()); - step.erase(step.begin()); + if(step.size() != 0) + step.erase(step.begin()); // assert step == 1 for (int i=0; i<(int)step.size(); i++)