法师大人 2023-03-30 18:08 采纳率: 100%
浏览 24
已结题

为什么用colab训练yolov5模型非常慢？

为什么用colab训练yolov5模型非常慢？

试了试网上的解决方法，我已经把整个项目包括数据集都拷贝到服务器的本地了，batch_size=32,num_work=8
训练速度还是非常慢，二十多分钟才能跑一轮

img

img

这是资源占用情况

img

  • 写回答

2条回答 默认 最新

  • 法师大人 2023-04-02 23:47
    关注

    已解决，因为我自己对yolov5模型做了些改动，除了原来要求的数据只要我的模型还有一组额外的输入，而这组输入我没有使用yolov5中的cache来读取，而是简单的使用im.read()，将我的数据也改用cache读取后训练速度提升很多

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题？

问题事件

  • 系统已结题 4月10日
  • 已采纳回答 4月2日
  • 创建了问题 3月30日