Refactored train_utils.py

2 jobs for master in 73 minutes and 59 seconds (queued for 1 second)