摘要:
很多新手在使用yolov5训练模型的时候会出现爆内存和爆显存问题,一般就是由于worker和batch_size参数设置过大有关,参考如下是解决的方法。 一、workers train.py中关于workers设置代码如下: workers是指数据装载时cpu所使用的线程数,默认为8,但是按照默认的 阅读全文
posted @ 2024-01-24 16:15
txwtech
阅读(1826)
评论(1)
推荐(0)
摘要:
yolov5介绍: https://blog.csdn.net/weixin_43334693/article/details/129312409 翻译 搜索 复制 阅读全文
posted @ 2024-01-24 11:43
txwtech
阅读(42)
评论(1)
推荐(0)

浙公网安备 33010602011771号