Add model metrics to the training summary.
Add model metrics to the training summary.

PiperOrigin-RevId: 307914609

tombstone/models,tombstone/models,tombstone/models,tombstone/models,tombstone/models,tombstone/models