改进yolov7时出现CUDA out of memory问题
企业开发
2023-04-08 06:46:08
阅读次数: 0
1.改batch-size,可以缩小为8,4,2等。
如果还不行
2.改workers,改为4,2,1,等。
如果还不行
3.在common中改小head的数量,可以改为2.
如果还不行
4.改img-size,可以将img-size改为【320,320】
转载自blog.csdn.net/weixin_44280747/article/details/128038375