search

PaddlePaddle / PaddleYOLO

🚀🚀🚀 YOLO series of PaddlePaddle implementation, PP-YOLOE+, RT-DETR, YOLOv5, YOLOv6, YOLOv7, YOLOv8, YOLOv10, YOLOX, YOLOv5u, YOLOv7u, YOLOv6Lite, RTMDet and so on. 🚀🚀🚀
https://github.com/PaddlePaddle/PaddleYOLO
GNU General Public License v3.0
554 stars 133 forks source link

use resnet_unit/yolo_unit fusion in yolov5 for xpu #152

Closed zhangyk0314 closed 7 months ago

zhangyk0314 commented 1 year ago

[Do Not Merge] 1、对csp_darknet backbone中的BaseConv进行fusion，通过YOLOv5_Fusion_ResnetUnit和YOLOv5_Fusion_YoloUnit环境变量进行控制，各自控制使用的fusion大算子，不开启则走默认的实现。 2、对yolov5_reader.yml文件中的use_shared_memory进行更改，该配置对模型训练性能会有很大影响。 3、AMP模式新增XPU后端支持。

nemonameless commented 7 months ago

您好，如还需提交PR，还请拉取最新代码，谢谢。