Added reporting for training runs.