据我所知,我正在使用本教程训练自己的对象检测器(https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/training.html),它没有教我们如何停止或何时停止转换工作。你们能帮我这个忙吗?我训练模型将近24小时,我的总损失约为2。
答案 0 :(得分:1)
损失是一个相对值,因为它与模型的准确性没有直接的关系,例如准确性,因此2的值并不能提供太多的见解。您可以查看损失是否正在减少,如果损失正在减少,则可以继续训练模型以执行更多步骤。
如果您的问题是如何设置时期数。这些配置将在* .config文件中完成。您可以编辑配置文件以更改批处理大小和步骤数的值。
训练的时期数=(训练集中的图像数/批次大小)*步数
*一个纪元是整个Entire数据集仅通过神经网络一次或一次通过