From 1377e6a08e7a4056124c4420d649fcfb1c116a9b Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Wed, 7 Dec 2022 12:51:00 +0800 Subject: [PATCH] Fix issue due to enable_onnx_checker only available in torch 0.9 Signed-off-by: Jael Gu --- auto_transformers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/auto_transformers.py b/auto_transformers.py index de10457..7e51693 100644 --- a/auto_transformers.py +++ b/auto_transformers.py @@ -115,7 +115,7 @@ class AutoTransformers(NNOperator): }, opset_version=13, do_constant_folding=True, - enable_onnx_checker=True, + # enable_onnx_checker=True, ) except Exception as e: print(e, '\nTrying with 2 outputs...')