netrans/bin/tools/onnx_visualize.py

9 lines
232 B
Python

import sys
from onnx.tools.net_drawer import main
# help --input xxx.onnx --output xxx.dot
main()
# The generate dot file, you can use graph viz editor to read it.
# graph viz editor can get from http://www.graphviz.org/download/