How to specify which metric to use for earlystopping?
For example, I define a custom compute_metrics which returns a dict including rouge and bleu. How to tell early stopping callback to stop training based on bleu?Thank you!
2021-10-22T11:46:39Z
You can specifiy in the fieldmetric_for_best_model(should be the name of the key in the dictionary returned by yourcompiute_metrics).
2021-10-22T12:04:31Z
Thank you! I will give it a try.