Fix flakiness of ReduceLROnPlateau tests (keras-team#135)

freedomtan · May 10, 2023 · eb09131 · eb09131
1 parent 6be0e2a
commit eb09131
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/keras_core/callbacks/reduce_lr_on_plateau_test.py b/keras_core/callbacks/reduce_lr_on_plateau_test.py
@@ -34,7 +34,7 @@ def setUp(self):
 
     def test_reduces_lr_with_model_fit(self):
         reduce_lr = callbacks.ReduceLROnPlateau(
-            patience=1, factor=0.1, monitor="val_loss", min_delta=10
+            patience=1, factor=0.1, monitor="val_loss", min_delta=100
         )
 
         self.model.fit(
@@ -49,7 +49,7 @@ def test_reduces_lr_with_model_fit(self):
 
     def test_throws_when_optimizer_has_schedule(self):
         reduce_lr = callbacks.ReduceLROnPlateau(
-            patience=1, factor=0.1, monitor="val_loss", min_delta=10
+            patience=1, factor=0.1, monitor="val_loss", min_delta=100
         )
 
         self.model.compile(
@@ -75,7 +75,7 @@ def test_throws_when_optimizer_has_schedule(self):
 
     def test_verbose_logging(self):
         reduce_lr = callbacks.ReduceLROnPlateau(
-            patience=1, factor=0.1, monitor="val_loss", min_delta=10, verbose=1
+            patience=1, factor=0.1, monitor="val_loss", min_delta=100, verbose=1
         )
         io_utils.disable_interactive_logging()
 
@@ -111,7 +111,11 @@ def test_honors_min_lr(self):
 
     def test_cooldown(self):
         reduce_lr = callbacks.ReduceLROnPlateau(
-            patience=1, factor=0.1, monitor="val_loss", min_delta=10, cooldown=2
+            patience=1,
+            factor=0.1,
+            monitor="val_loss",
+            min_delta=100,
+            cooldown=2,
         )
 
         self.model.fit(