During training checkpoints are being saved, also the last version of the trained model, which might not be the best model. So there should be an option to save the best model as well.

Prerequisites

Please answer the following question for yourself before submitting an issue.

• I checked to make sure that this feature has not been requested already.

1. The entire URL of the file you are using

https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/tf2_detection_zoo.md

2. Describe the feature you request

During training model checkpoints are being saved, also the last version of the trained model, which might not be the best model. So there should be an option to save the best model as well. I know of the tf.estimator.BestExporter estimator, but how can I use that with the latest and official model_main_tf2.py script of the object detection API?

Moreover, until there is such a solution, how can one get the best model during or after the training via the object detection API?

3. Additional context

I am following the official object detection API docs, thus using the model_main_tf2.py script, so a solution should be based on that script too, if possible.

4. Are you willing to contribute it? (Yes or No)

In theory yes, but I am afraid that I don't know the TF object detection API well enough to be able to help.