Fix issue due to enable_onnx_checker only available in torch 0.9

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
2 years ago · 1377e6a08e
1 changed files with 1 additions and 1 deletions
--- 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...')