"""Implements Early stopping for PyTorch model."""