早停 Tensorflow object 检测 API

Question

正如标题所暗示的那样，无论如何要在 TF object 检测 API 上实现提前停止？

i read in this thread https://github.com/tensorflow/models/issues/5887 on the github of Object detection that there's a repo https://github.com/hongym7/early_stopping for early stopping. 但这是 TF1 的旧版本，我使用 TF2

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

这是训练 model 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了解更多详情