GitLab

* refac: docstrings in training_utils.py

* fix: manual edits

* run make style

* add docstring at cast_training_params

---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>