改进yolov7时出现CUDA out of memory问题

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