validation loss fix