提交 fbc1245f 编写于 作者: C channingss

delete check version of paddle

上级 35fba2f5
...@@ -142,12 +142,12 @@ def onnx2paddle(model_path, save_dir): ...@@ -142,12 +142,12 @@ def onnx2paddle(model_path, save_dir):
try: try:
import torch import torch
version = torch.__version__ version = torch.__version__
if '1.1.0' not in version: if '1.2.0' not in version:
print("torch==1.1.0 is required") print("torch==1.2.0 is required")
return return
except: except:
print( 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 return
try: try:
...@@ -190,14 +190,6 @@ def main(): ...@@ -190,14 +190,6 @@ def main():
x2paddle.__version__)) x2paddle.__version__))
return 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.framework is not None, "--framework is not defined(support tensorflow/caffe/onnx)"
assert args.save_dir is not None, "--save_dir is not defined" 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.
先完成此消息的编辑!
想要评论请 注册