Reduce console spam when using the KerasMetricCallback (#18202)
Reduce console spam when using the KerasMetricCallback (#18202)

* Reduce console spam when using the KerasMetricCallback

* Switch to predict_on_batch to improve performance
huggingface/transformers,huggingface/transformers,huggingface/transformers,huggingface/transformers