提前停止 TensorFlow 对象检测 API

Question

正如标题所暗示的那样，是否有在 TF 对象检测 API 上实现提前停止？

我在对象检测的 github 上的这个线程 https://github.com/tensorflow/models/issues/5887 中读到，有一个用于提前停止的 repo https://github.com/hongym7/early_stopping。但这是 TF1 的旧版本，我使用的是 TF2

有没有办法在 model_main_tf2.py 上实现提前停止？我花了几个小时阅读代码，但我没有看到任何地方可以实现提前停止。

这是训练模型的脚本链接 https://github.com/tensorflow/models/blob/master/research/object_detection/model_main_tf2.py

Answer 1

用 tf.keras 重写它并使用 tf.keras.callbacks.EarlyStopping 。阅读https://www.tensorflow.org/api_docs/python/tf/keras/callbacks/EarlyStopping了解更多详情