Browse Source

Tiny fix mxnet2ncnn tool (#1691)

tags/20200413
mengfu188 GitHub 6 years ago
parent
commit
d06606b0cd
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      tools/mxnet/mxnet2ncnn.cpp

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

@@ -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++)


Loading…
Cancel
Save