diff --git a/paddleslim/common/load_model.py b/paddleslim/common/load_model.py index 12d0e46f470c45a1fdc30369c1319ea30e246d8a..ced6eaf3327cb5b74f53823f52010684187596ce 100644 --- a/paddleslim/common/load_model.py +++ b/paddleslim/common/load_model.py @@ -218,11 +218,11 @@ def export_onnx(model_dir, try: import paddle2onnx version = paddle2onnx.__version__ - if version != '1.0.1': - os.system('python -m pip install -U paddle2onnx==1.0.1') + if version < '1.0.1': + os.system('python -m pip install -U paddle2onnx==1.0.3') except: from pip._internal import main - main(['install', 'paddle2onnx==1.0.1']) + main(['install', 'paddle2onnx==1.0.3']) import paddle2onnx paddle2onnx.command.c_paddle_to_onnx( model_file=os.path.join(model_dir, model_filename),