早停 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了解更多詳情