为什么用colab训练yolov5模型非常慢?
试了试网上的解决方法,我已经把整个项目包括数据集都拷贝到服务器的本地了,batch_size=32,num_work=8
训练速度还是非常慢,二十多分钟才能跑一轮
这是资源占用情况
为什么用colab训练yolov5模型非常慢?
试了试网上的解决方法,我已经把整个项目包括数据集都拷贝到服务器的本地了,batch_size=32,num_work=8
训练速度还是非常慢,二十多分钟才能跑一轮
这是资源占用情况
已解决,因为我自己对yolov5模型做了些改动,除了原来要求的数据只要我的模型还有一组额外的输入,而这组输入我没有使用yolov5中的cache来读取,而是简单的使用im.read(),将我的数据也改用cache读取后训练速度提升很多