fix off by one error (see #2289) epoch starts at 0 now

Lightning-AI · Jun 22, 2020 · c5330ed · c5330ed
1 parent 1a39e1d
commit c5330ed
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/callbacks/test_early_stopping.py b/tests/callbacks/test_early_stopping.py
@@ -66,9 +66,9 @@ def on_train_end(self, trainer, pl_module):
 
 
 @pytest.mark.parametrize('loss_values, patience, expected_stop_epoch', [
-    ([6, 5, 5, 5, 5, 5], 3, 5),
-    ([6, 5, 4, 4, 3, 3], 1, 4),
-    ([6, 5, 6, 5, 5, 5], 3, 5),
+    ([6, 5, 5, 5, 5, 5], 3, 4),
+    ([6, 5, 4, 4, 3, 3], 1, 3),
+    ([6, 5, 6, 5, 5, 5], 3, 4),
 ])
 def test_early_stopping_patience(loss_values, patience, expected_stop_epoch):
     """Test to ensure that early stopping is not triggered before patience is exhausted."""