提前停止Tensorflow对象检测API



正如标题所建议的那样,是否应该在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了解更多细节

相关内容

  • 没有找到相关文章

最新更新