Explorar o código

set the version of paddle2onnx to 0.4

Channingss %!s(int64=4) %!d(string=hai) anos
pai
achega
a1f4d05325
Modificáronse 2 ficheiros con 2 adicións e 2 borrados
  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 
     except:
         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
     if opset_version == 10 and model.__class__.__name__ == "YOLOv3":
         logging.warning(

+ 1 - 1
requirements.txt

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