VGGNet微调：图像大小，时间和时代

Question

我是深度学习的新手，我希望有一些澄清来理解深度学习。

我想使用VGGNet训练我的CNN模型，我应该使用相同图像尺寸的VGGNet 224x224 输入图像进行微调吗？

如何确定纪元的数量？

如果图像数量约为5000张图像且GPU Nvidia gtx 1070，如何训练模型？

请帮我找到答案。 谢谢

Answer 1

IT

我认为纪元的数量是你遍历整个数据的次数。实际上，时代的数量无助于您决定网络何时何地会聚。这在很大程度上取决于您的算法和数据本身。所以可以忽略它。

您最重要的目标是找到具有最佳准确度的模型。根据我的经验，让机器运行20个时代。然后绘制损失精度图并选择模型。

例如，在下图中。我经常选择从红色范围保存的模型。这有助于我避免过度贴合。 （图像从互联网复制）

希望有所帮助。