Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

ONNX export: Slice op - Handle None value for ends #14942

Merged
merged 7 commits into from
Feb 20, 2020

Conversation

vandanavk
Copy link
Contributor

@vandanavk vandanavk commented May 14, 2019

Description

Fixes #14875

Checklist

Essentials

Please feel free to remove inapplicable items for your PR.

  • The PR title starts with [MXNET-$JIRA_ID], where $JIRA_ID refers to the relevant JIRA issue created (except PRs with tiny changes)
  • Changes are complete (i.e. I finished coding on this PR)
  • All changes have test coverage:
  • Unit tests are added for small changes to verify correctness (e.g. adding a new operator)
  • Nightly tests are added for complicated/long-running ones (e.g. changing distributed kvstore)
  • Build tests will be added for build configuration changes (e.g. adding a new build option with NCCL)
  • Code is well-documented:
  • For user-facing API changes, API doc string has been updated.
  • For new C++ functions in header files, their functionalities and arguments are documented.
  • For new examples, README.md is added to explain the what the example does, the source of the dataset, expected performance on test set and reference to the original paper if applicable
  • Check the API doc at http://mxnet-ci-doc.s3-accelerate.dualstack.amazonaws.com/PR-$PR_ID/$BUILD_ID/index.html
  • To the my best knowledge, examples are either not affected by this change, or have been fixed to be compatible with this change

Changes

  • Handle ends=None in Slip_axis export to ONNX

Comments

  • All ONNX export tests passed

@vandanavk vandanavk requested a review from szha as a code owner May 14, 2019 17:54
@vandanavk
Copy link
Contributor Author

@mxnet-label-bot add [ONNX, pr-awaiting-review]

@marcoabreu marcoabreu added ONNX pr-awaiting-review PR is waiting for code review labels May 14, 2019
@Roshrini
Copy link
Member

@vandanavk Thanks for fixing this. Can you please add unit test or model test?

@karan6181
Copy link
Contributor

@vandanavk Did you get the chance to add the unit test or model test ? Thanks!

@piyushghai
Copy link
Contributor

@vandanavk Gentle ping....

@vandanavk
Copy link
Contributor Author

Working on the test. Will have an update soon.

@karan6181
Copy link
Contributor

@vandanavk any updates on the progress?

@ntomer
Copy link

ntomer commented Aug 1, 2019

I am awaiting the changes :)

@vandanavk
Copy link
Contributor Author

This PR is ready for review. Awaiting confirmation from the user too

@ntomer
Copy link

ntomer commented Aug 22, 2019

Any word on when this might pass review?

@vandanavk
Copy link
Contributor Author

@Roshrini @anirudhacharya @zhreshold for review

@vandanavk
Copy link
Contributor Author

Based on offline discussion with @Roshrini, making use of ONNX's "slice_end_out_of_bounds" test instead of writing a custom test

@vandanavk
Copy link
Contributor Author

All review comments addressed. Awaiting further review comments or approval.

@vandanavk
Copy link
Contributor Author

@zhreshold for review/merge

@vandanavk
Copy link
Contributor Author

@mxnet-label-bot update [ONNX, pr-awaiting-merge]

@lanking520 lanking520 added pr-awaiting-merge Review and CI is complete. Ready to Merge and removed pr-awaiting-review PR is waiting for code review labels Jan 17, 2020
@anirudhacharya anirudhacharya merged commit 9dcf71d into apache:master Feb 20, 2020
@chouxianyu
Copy link

Environment:

  • Windows10
  • Python3.7
  • CPU

Descroption:
I met this problem.
And I tried the solution in 9dcf71d, found a new bug.

Traceback (most recent call last):
File "C:\Program Files\PyCharm2018.1.4\helpers\pydev\pydevd.py", line 1664, in
main()
File "C:\Program Files\PyCharm2018.1.4\helpers\pydev\pydevd.py", line 1658, in main
globals = debugger.run(setup['file'], None, None, is_module)
File "C:\Program Files\PyCharm2018.1.4\helpers\pydev\pydevd.py", line 1068, in run
pydev_imports.execfile(file, globals, locals) # execute the script
File "C:\Program Files\PyCharm2018.1.4\helpers\pydev_pydev_imps_pydev_execfile.py", line 18, in execfile
exec(compile(contents+"\n", file, 'exec'), glob, loc)
File "E:/Study/Programming/Code/FWWBA05/MXNet_Helmet_Detection/ONNXTest.py", line 2, in
export_model('./symbols/darknet53-symbol.json', './symbols/darknet53-0000.params', [(4, 3, 416, 416)], verbose=True)
File "D:\WorkingSoftware\Anaconda3\lib\site-packages\mxnet\contrib\onnx\mx2onnx\export_model.py", line 83, in export_model
verbose=verbose)
File "D:\WorkingSoftware\Anaconda3\lib\site-packages\mxnet\contrib\onnx\mx2onnx\export_onnx.py", line 253, in create_onnx_graph_proto
idx=idx
File "D:\WorkingSoftware\Anaconda3\lib\site-packages\mxnet\contrib\onnx\mx2onnx\export_onnx.py", line 90, in convert_layer
raise AttributeError("No conversion function registered for op type %s yet." % op)
AttributeError: No conversion function registered for op type _arange yet.

anirudh2290 pushed a commit to anirudh2290/mxnet that referenced this pull request May 29, 2020
* ONNX export: Slice op - Handle None value for ends
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
ONNX pr-awaiting-merge Review and CI is complete. Ready to Merge
Projects
None yet
Development

Successfully merging this pull request may close these issues.

MXNet to ONNX export bug
9 participants