Revalidation doesn't always use the best model

[pgagarinov]

Might be a issue with PyTorch-Lightning not loading the best model for some reason, needs further investigation