YOLOv4训练提示 Out of memory!
网上的办法如果都解决不了，试试我的这个。
训练时候修改你自己的cfg文件中height和weight，往小改，官方为416，我改成960报以上错误。
height=960
weight=960
根本原因是，输入网络的图片尺寸越大，卷积核计算出最后的tensor的维度会过高，可以理解为参数爆炸，所以YOLO也不能处理过高分辨率的图片。
Think：想一个办法改进？
YOLOv4训练提示 Out of memory!
猜你喜欢
转载自blog.csdn.net/weixin_40557160/article/details/108961229
今日推荐
周排行