Pin paddlepaddle<3.0.0 to avoid bug in latest release (#13548)
* Update export.py Signed-off-by: Muhammad Rizwan Munawar <muhammadrizwanmunawar123@gmail.com> * Pin "paddlepaddle<3.0.0" to resolve export bug (#13550) --------- Signed-off-by: Muhammad Rizwan Munawar <muhammadrizwanmunawar123@gmail.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
1 parent
61c22e2d20
commit
3bc8a531fe
2 files changed
+2
-2
No files matched your search
@@ -510,7 +510,7 @@ def export_paddle(model, im, file, metadata, prefix=colorstr("PaddlePaddle:")):
|
||||
$ pip install paddlepaddle x2paddle
|
||||
```
|
||||
"""
|
||||
check_requirements(("paddlepaddle", "x2paddle"))
|
||||
check_requirements(("paddlepaddle<3.0.0", "x2paddle"))
|
||||
import x2paddle
|
||||
from x2paddle.convert import pytorch2paddle
|
||||
|
||||
|
||||
+1
-1
@@ -646,7 +646,7 @@ class DetectMultiBackend(nn.Module):
|
||||
raise NotImplementedError("ERROR: YOLOv5 TF.js inference is not supported")
|
||||
elif paddle: # PaddlePaddle
|
||||
LOGGER.info(f"Loading {w} for PaddlePaddle inference...")
|
||||
check_requirements("paddlepaddle-gpu" if cuda else "paddlepaddle")
|
||||
check_requirements("paddlepaddle-gpu" if cuda else "paddlepaddle<3.0.0")
|
||||
import paddle.inference as pdi
|
||||
|
||||
if not Path(w).is_file(): # if not *.pdmodel
|
||||
|
||||
Reference in new issue
Block a user