update test

Lightning-AI · Oct 13, 2021 · 40cecaa · 40cecaa
1 parent 03e5a6e
commit 40cecaa
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/callbacks/test_lr_monitor.py b/tests/callbacks/test_lr_monitor.py
@@ -535,6 +535,9 @@ def finetune_function(self, pl_module, epoch: int, optimizer, opt_idx: int):
     model.training_epoch_end = None
     trainer.fit(model)
 
+    expected = [0.1, 0.1, 0.1, 0.1, 0.1]
+    assert lr_monitor.lrs["lr-Adam-2"] == expected
+
     expected = [0.1, 0.05, 0.025, 0.0125, 0.00625]
     assert lr_monitor.lrs["lr-Adam/pg1"] == expected