catch onnx checker error

apache · May 15, 2019 · bc3cbea · bc3cbea
1 parent f6e9ef1
commit bc3cbea
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py
@@ -1140,6 +1140,10 @@ def from_onnx(model,
             checker.check_model(model)
     except ImportError:
         pass
+    except onnx.onnx_cpp2py_export.checker.ValidationError as e:
+        import warnings
+        # the checker is a bit violent about errors, so simply print warnings here
+        warnings.warn(e)
     g = GraphProto(shape, dtype)
     graph = model.graph
     try: