Fixed Quitting after 5 min of training