正如标题所建议的那样,是否应该在TF对象检测API上实现早期停止?
我在这个线程读https://github.com/tensorflow/models/issues/5887在github上的对象检测,有一个回购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
在tf.keras
中重写,使用tf.keras.callbacks.EarlyStopping
。阅读https://www.tensorflow.org/api_docs/python/tf/keras/callbacks/EarlyStopping了解更多细节