我该如何停止tensorflow中的培训工作?

时间:2019-10-12 17:33:50

标签: python tensorflow

据我所知,我正在使用本教程训练自己的对象检测器(https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/training.html),它没有教我们如何停止或何时停止转换工作。你们能帮我这个忙吗?我训练模型将近24小时,我的总损失约为2。

1 个答案:

答案 0 :(得分:1)

损失是一个相对值,因为它与模型的准确性没有直接的关系,例如准确性,因此2的值并不能提供太多的见解。您可以查看损失是否正在减少,如果损失正在减少,则可以继续训练模型以执行更多步骤。

如果您的问题是如何设置时期数。这些配置将在* .config文件中完成。您可以编辑配置文件以更改批处理大小和步骤数的值。

训练的时期数=(训练集中的图像数/批次大小)*步数

*一个纪元是整个Entire数据集仅通过神经网络一次或一次通过