Refac training utils.py (#9815)

Commit

323 days ago

Refac training utils.py (#9815) * Refac training utils.py * quality --------- Co-authored-by: sayakpaul <spsayakpaul@gmail.com>