我正在使用YOLOv3进行对象检测的训练，但当我将batch_size设置为大于1时，遇到了CUDA内存不足的问题，于是我在谷歌上搜索了其他解决方案，发现这取决于我的GPU（GTX 1070 8G）。

可能是epoch的数量过多，需要进行优化。

或许应该减少epoch的数量？我有1k张图片用于训练，200张图片用于验证。

为了避免过拟合，我应该设置多少个epoch才是最佳的？

回答：

你的模型是否过拟合并不取决于你设置的epoch数量……
由于你已经在数据中进行了验证集的分割，请确保你的训练损失 – 验证损失 或 训练准确率 – 验证准确率大致相同。这将确保你的模型没有过拟合。