diff --git a/x2paddle/convert.py b/x2paddle/convert.py index 5e23b010c4b44efb9c698388226001251d901c1d..24dd0fc837980d5397ef2d7a5d4bbac12b83809f 100644 --- a/x2paddle/convert.py +++ b/x2paddle/convert.py @@ -149,7 +149,9 @@ def onnx2paddle(model_path, save_dir): try: import onnx version = onnx.version.version - if version < '1.6.0': + v0, v1, v2 = version.split('.') + version_sum = int(v0) * 100 + int(v1) * 10 + int(v2) + if version_sum < 160: print("[ERROR] onnx>=1.6.0 is required") return except: