Update `trainer_pt_utils.py` docstrings for consistency (#36912)
* Update trainer_pt_utils.py
* update docstrings trainer_pt_utils.py for consistency
* Update src/transformers/trainer_pt_utils.py
---------
Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>