Improve _train_n_steps logging message.
Improve _train_n_steps logging message.

PiperOrigin-RevId: 320855556

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