【发布时间】:2020-03-02 06:51:52
【问题描述】:
我正在训练一个卷积神经网络,使用 tensorboard 记录训练趋势。
结果看起来有点奇怪,因为它几乎是一条直线,但在我看来它应该像 logcurve。
这有一些原因,但我不太确定。
- 小数据集:我得到10个类,训练集的大小是432张形状为(256, 512)的图像。虽然我已经做了一些增强工作
- 浅层网络:因为我在家里只有一个 2GB 内存的 GPU。所以网络只有 4 个卷积层和 2 个 FN 层。(是的病毒诅咒,为中国祈祷!)
所以谁能告诉我这是什么原因?我该如何解决?
【问题讨论】:
-
也许你应该训练更长的时间?该模型显然还没有收敛。
标签: tensorflow computer-vision conv-neural-network