Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Create script in runner to convert opset versions #1

Merged
merged 1 commit into from
Mar 26, 2020

Conversation

MaxBanister
Copy link
Collaborator

Create a script convert_to_opset9.py that takes in an ONNX model of opset version 7 as the first argument on the command line and converts it to one of version 9. The new file has the same name as the old one with a "_op9" before the file extension.

@pranav-prakash pranav-prakash merged commit 80bbd3e into systolic Mar 26, 2020
@pranav-prakash
Copy link
Collaborator

LGTM, merged. Thanks!
P.S. although it's probably overkill for this, you might consider using pathlib to get the .stem of the filename (instead of [:-5]) and the wonderful argparse for parsing cmd args,

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants