Add validation tests after each epoch

We need to check if our model performs better after each epoch using the validation data