added a flag to allow more accurate compuation of training rmse results