yolov5显存溢出

运行train.py的时候总是现存溢出，然后修改了batch_size也无济于事，想问一下大家这个问题是怎么解决的呀？

解决方案：

不知道你这个问题是否已经解决, 如果还没有解决的话:

• 这篇博客: 参数详解：yolov7模型 训练部分（train.py）中的 6、batch-size 部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:
  

  parser.add_argument('--batch-size', type=int, default=16, help='total batch size for all GPUs')

  每批次的输入数据量；default=-1将时自动调节batchsize大小，default=16，代表一次输入16张图片进行训练。

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^

租个云GPU吧

2G显存基本上只能跑一跑yolov5n.pt,更大一些的yolov5s我估计你只能用小batch和小的imgs参数才行，至少640x640一个batch都有点悬的样子。