Merge pull request #4 from wolframalpha/master

Update pytorchtools.py
Bjarten · Aug 27, 2019 · effbcce · effbcce
2 parents 36cff88 + 780b1dc
commit effbcce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorchtools.py b/pytorchtools.py
@@ -28,7 +28,7 @@ def __call__(self, val_loss, model):
         if self.best_score is None:
             self.best_score = score
             self.save_checkpoint(val_loss, model)
-        elif score < self.best_score - delta:
+        elif score < self.best_score - self.delta:
             self.counter += 1
             print(f'EarlyStopping counter: {self.counter} out of {self.patience}')
             if self.counter >= self.patience: