Save model after each epoch & add early stopping
Add early stopping to not continue with training if the validation loss has converged (and starts to increase again) + save model after each epoch and finally choose the best one
Add early stopping to not continue with training if the validation loss has converged (and starts to increase again) + save model after each epoch and finally choose the best one