Forráskód Böngészése

set the version of paddle2onnx to 0.4

Channingss 4 éve
szülő
commit
a1f4d05325
2 módosított fájl, 2 hozzáadás és 2 törlés
  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