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
|
$ pip install paddlepaddle x2paddle
|
||||||
```
|
```
|
||||||
"""
|
"""
|
||||||
check_requirements(("paddlepaddle", "x2paddle"))
|
check_requirements(("paddlepaddle<3.0.0", "x2paddle"))
|
||||||
import x2paddle
|
import x2paddle
|
||||||
from x2paddle.convert import pytorch2paddle
|
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")
|
raise NotImplementedError("ERROR: YOLOv5 TF.js inference is not supported")
|
||||||
elif paddle: # PaddlePaddle
|
elif paddle: # PaddlePaddle
|
||||||
LOGGER.info(f"Loading {w} for PaddlePaddle inference...")
|
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
|
import paddle.inference as pdi
|
||||||
|
|
||||||
if not Path(w).is_file(): # if not *.pdmodel
|
if not Path(w).is_file(): # if not *.pdmodel
|
||||||
|
|||||||
Reference in new issue
Block a user