未验证 提交 38f6f578 编写于 作者: G Guanghua Yu 提交者: GitHub

fix paddle2onnx version when export new format quant model (#1458)

上级 bb1f712c
...@@ -217,11 +217,11 @@ def export_onnx(model_dir, ...@@ -217,11 +217,11 @@ def export_onnx(model_dir,
try: try:
import paddle2onnx import paddle2onnx
version = paddle2onnx.__version__ version = paddle2onnx.__version__
if version != '1.0.0rc4': if version != '1.0.1':
os.system('python -m pip install -U paddle2onnx==1.0.0rc4') os.system('python -m pip install -U paddle2onnx==1.0.1')
except: except:
from pip._internal import main from pip._internal import main
main(['install', 'paddle2onnx==1.0.0rc4']) main(['install', 'paddle2onnx==1.0.1'])
import paddle2onnx import paddle2onnx
paddle2onnx.command.c_paddle_to_onnx( paddle2onnx.command.c_paddle_to_onnx(
model_file=os.path.join(model_dir, model_filename), model_file=os.path.join(model_dir, model_filename),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册