diff --git a/utils/tools.py b/utils/tools.py index 882c239..6f29785 100644 --- a/utils/tools.py +++ b/utils/tools.py @@ -28,7 +28,7 @@ def __init__(self, patience=7, verbose=False, delta=0): self.counter = 0 self.best_score = None self.early_stop = False - self.val_loss_min = np.Inf + self.val_loss_min = np.inf self.delta = delta def __call__(self, val_loss, model, path):