fix train and val error reporting