Skip to content

Instantly share code, notes, and snippets.

@sandeepkumar-skb

sandeepkumar-skb/onnx_checker_printer.py

Forked from mkolod/onnx_checker_printer.py
Created June 25, 2020 01:55
Show Gist options
  • Save sandeepkumar-skb/a8ed079a58c76ede455368bd12db60c5 to your computer and use it in GitHub Desktop.
Save sandeepkumar-skb/a8ed079a58c76ede455368bd12db60c5 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
onnx_checker_printer.py
import onnx
import sys
name = sys.argv[1]
model = onnx.load(name)
onnx.checker.check_model(model)
print(onnx.helper.printable_graph(model.graph))
@sandeepkumar-skb
Copy link
Author

sandeepkumar-skb commented Jun 26, 2020 •
edited
Loading

Install the onnx module by running the following:
conda install -c conda-forge onnx

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment