From fbc1245f85d28b4e31b5368d6c43a3424fed0792 Mon Sep 17 00:00:00 2001 From: channingss Date: Thu, 5 Sep 2019 12:02:10 +0800 Subject: [PATCH] delete check version of paddle --- x2paddle/convert.py | 14 +++----------- 1 file changed, 3 insertions(+), 11 deletions(-) diff --git a/x2paddle/convert.py b/x2paddle/convert.py index 5ef5065..1dc83e5 100644 --- a/x2paddle/convert.py +++ b/x2paddle/convert.py @@ -142,12 +142,12 @@ def onnx2paddle(model_path, save_dir): try: import torch version = torch.__version__ - if '1.1.0' not in version: - print("torch==1.1.0 is required") + if '1.2.0' not in version: + print("torch==1.2.0 is required") return except: print( - "we use caffe2 to inference graph, please use \"pip install torch==1.1.0\"." + "we use caffe2 to inference graph, please use \"pip install torch==1.2.0\"." ) return try: @@ -190,14 +190,6 @@ def main(): x2paddle.__version__)) return - try: - import paddle - v0, v1, v2 = paddle.__version__.split('.') - if int(v0) != 1 or int(v1) < 5: - print("paddlepaddle>=1.5.0 is required") - return - except: - print("paddlepaddle not installed, use \"pip install paddlepaddle\"") assert args.framework is not None, "--framework is not defined(support tensorflow/caffe/onnx)" assert args.save_dir is not None, "--save_dir is not defined" -- GitLab