jiazhihao / TASO

The Tensor Algebra SuperOptimizer for Deep Learning
Apache License 2.0
687 stars 90 forks source link

Unsupported ONNX operator: Squeeze #7

Closed olivia111 closed 4 years ago

olivia111 commented 4 years ago

Traceback (most recent call last): File "examples/test_onnx.py", line 4, in graph = taso.load_onnx("/home/taso/yolov3.onnx") File "/home/.local/lib/python3.7/site-packages/taso-0.1.0-py3.7-linux-x86_64.egg/taso/init.py", line 459, in load_onnx assert False, "Unsupported ONNX operator: {}".format(op.op_type) AssertionError: Unsupported ONNX operator: Squeeze Error in sys.excepthook: Traceback (most recent call last): File "/usr/lib/python3/dist-packages/apport_python_hook.py", line 63, in apport_excepthook from apport.fileutils import likely_packaged, get_recent_crashes File "/usr/lib/python3/dist-packages/apport/init.py", line 5, in from apport.report import Report File "/usr/lib/python3/dist-packages/apport/report.py", line 30, in import apport.fileutils File "/usr/lib/python3/dist-packages/apport/fileutils.py", line 23, in from apport.packaging_impl import impl as packaging File "/usr/lib/python3/dist-packages/apport/packaging_impl.py", line 24, in import apt File "/usr/lib/python3/dist-packages/apt/init.py", line 23, in import apt_pkg ModuleNotFoundError: No module named 'apt_pkg'

Original exception was: Traceback (most recent call last): File "examples/test_onnx.py", line 4, in graph = taso.load_onnx("/home/taso/yolov3.onnx") File "/home/.local/lib/python3.7/site-packages/taso-0.1.0-py3.7-linux-x86_64.egg/taso/init.py", line 459, in load_onnx assert False, "Unsupported ONNX operator: {}".format(op.op_type) AssertionError: Unsupported ONNX operator: Squeeze

jiazhihao commented 4 years ago

Support for Squeeze has been added.