Browse Source

set the version of paddle2onnx to 0.4

Channingss 4 years ago
parent
commit
a1f4d05325
2 changed files with 2 additions and 2 deletions
  1. 1 1
      paddlex/converter.py
  2. 1 1
      requirements.txt

+ 1 - 1
paddlex/converter.py

@@ -267,7 +267,7 @@ def export_onnx_model(model, save_file, opset_version=10):
         import paddle2onnx 
         import paddle2onnx 
     except:
     except:
         logging.error(
         logging.error(
-            "You need to install paddle2onnx first, pip install paddle2onnx")
+            "You need to install paddle2onnx first, pip install paddle2onnx==0.4")
     import paddle2onnx as p2o
     import paddle2onnx as p2o
     if opset_version == 10 and model.__class__.__name__ == "YOLOv3":
     if opset_version == 10 and model.__class__.__name__ == "YOLOv3":
         logging.warning(
         logging.warning(

+ 1 - 1
requirements.txt

@@ -6,7 +6,7 @@ pycocotools
 visualdl >= 2.0.0b
 visualdl >= 2.0.0b
 paddleslim == 1.1.1
 paddleslim == 1.1.1
 shapely
 shapely
-paddle2onnx
+paddle2onnx==0.4
 paddlepaddle-gpu
 paddlepaddle-gpu
 opencv-python
 opencv-python
 psutil
 psutil