提交 fbc1245f 编写于 作者: C channingss

delete check version of paddle

上级 35fba2f5
......@@ -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"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册