【问题标题】:why my training trend is a straight line?为什么我的训练趋势是一条直线?
【发布时间】:2020-03-02 06:51:52
【问题描述】:

我正在训练一个卷积神经网络,使用 tensorboard 记录训练趋势。

结果看起来有点奇怪,因为它几乎是一条直线,但在我看来它应该像 logcurve。

这有一些原因,但我不太确定。

  • 小数据集:我得到10个类,训练集的大小是432张形状为(256, 512)的图像。虽然我已经做了一些增强工作
  • 浅层网络:因为我在家里只有一个 2GB 内存的 GPU。所以网络只有 4 个卷积层和 2 个 FN 层。(是的病毒诅咒,为中国祈祷!)

所以谁能告诉我这是什么原因?我该如何解决?

【问题讨论】:

  • 也许你应该训练更长的时间?该模型显然还没有收敛。

标签: tensorflow computer-vision conv-neural-network


【解决方案1】:

感谢@xdurch0 的建议。

我之前得到结果的原因是我没有设置正确的训练 epoch 太小了(虽然我已经训练了 128 个 epoch)

现在我发布一张图表,其中包括两个培训记录以确认原因。

另外 使用keras.callbacks.EealyStopping 时应该非常谨慎,因为我们可以看到训练趋势是一个复杂的 zip-zap。否则我们会过早或过晚停止。

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2022-09-30
    • 1970-01-01
    • 2018-03-24
    • 1970-01-01
    • 1970-01-01
    • 2021-08-19
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多