晒月光12138 2022-11-21 16:37
浏览 723
已结题

yolov5在fusing layers停顿太久

笔记本运行YOLOv5,3050显卡,python3.8.0 torch-1.12.1 cuda113,能够正常运行检测,每张图像检测速度也很快,但是在代码运行后会“Fusing layers”这一行卡顿很久很久,估计有几分钟的时间,才可以正常运行检测。这里使用的是服务器上训练好的模型,服务器上就没有类似的问题,每次代码运行到这里通过的都很快。

YOLOv5 🚀 2022-9-11 Python-3.8.0 torch-1.12.1+cu113 CUDA:0 (GeForce RTX 3050 Laptop GPU, 3902MiB)

Fusing layers... 
Model summary: 213 layers, 7012822 parameters, 0 gradients

  • 写回答

0条回答 默认 最新

    报告相同问题?

    问题事件

    • 系统已结题 11月29日
    • 修改了问题 11月21日
    • 创建了问题 11月21日