Early Stopping Pytorch Lightning