Merge pull request #98 from lugen4ro/fix_numpy_attribute_error

Fix numpy error pertaining to np.Inf
tsugumi-sys · Jul 24, 2024 · afe4c80 · afe4c80
2 parents fd8a460 + f8f953c
commit afe4c80
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pipelines/utils/early_stopping.py b/pipelines/utils/early_stopping.py
@@ -28,7 +28,7 @@ def __init__(
         self.counter = 0
         self.best_score: Optional[float] = None
         self.early_stop = False
-        self.val_loss_min = np.Inf
+        self.val_loss_min = np.inf
         self.delta = delta
         self.model_filename = model_save_path
         if not model_save_path.endswith(".pt"):