【发布时间】:2020-02-09 21:30:33
【问题描述】:
我正在使用本教程来训练我自己的对象检测器 (https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/training.html),据我所知,它并没有教我们如何停止或何时停止训练工作。你们能帮我解决这个问题吗?我训练我的模型将近 24 小时,我的总损失约为 2。
【问题讨论】:
标签: python tensorflow
我正在使用本教程来训练我自己的对象检测器 (https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/training.html),据我所知,它并没有教我们如何停止或何时停止训练工作。你们能帮我解决这个问题吗?我训练我的模型将近 24 小时,我的总损失约为 2。
【问题讨论】:
标签: python tensorflow
损失是一个相对值,因为它与模型的准确性没有直接相关性,因此值 2 并不能提供太多洞察力。如果损失减少,您可以查看损失是否减少,您可以继续训练模型以进行更多步数。
如果您的问题是如何设置 epoch 数。这些配置将在 *.config 文件中完成。您可以编辑配置文件以更改批量大小和步数的值。
训练的 epoch 数 =(训练集中的图像数 / 批量大小)*num_steps
*一个时期是指整个数据集仅通过神经网络向前和向后传递一次
【讨论】: