Pytorch Early Stopping Example