训练1k图像需要多少个历元

2024-04-18 18:31:12 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在进行使用yolov3检测对象的培训,但当我设置batch_size>;1它导致我的cuda内存不足,所以我在google中搜索了另一个解决方案,发现它依赖于我的GPU(GTX 1070 8G)

可能是历元数高,需要优化

也许历元数应该减少?用于培训1k图像和200张pic进行验证

我应该设置什么样的最佳时期来避免过度装修


Tags: 对象图像gtsizegpubatchgoogle解决方案
1条回答
网友
1楼 · 发布于 2024-04-18 18:31:12

您的模型是否过合适并不取决于您设置的年代数….
由于您已经在数据中进行了val分割,请确保您的train loss-val loss或train acc-val acc几乎相同。这将确保您的模型不会过度拟合

相关问题 更多 >