mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Pin "paddlepaddle<3.0.0" to resolve export bug (#13550)

pull/13548/head
Glenn Jocher 2025-03-27 08:55:55 +01:00 committed by GitHub
parent ad450f09c6
commit 6993b75af2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models/common.py
View File

@ -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