Skip to content

Commit 8004fef

Browse files
authored
Suppress the PyTorch warning when verifying LR Scheduler (#1356)
As title.
1 parent 6b11290 commit 8004fef

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/unit_tests/test_lr_scheduler.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,11 @@ def setUp(self):
2020
self.model = torch.nn.Linear(10, 10)
2121
# Create an optimizer
2222
self.optimizer = Adam(self.model.parameters(), lr=0.1)
23+
24+
# We don't actually call `optimizer.step()` which will cause a warning
25+
# from PyTorch. Avoid the warnings that may confuse people.
26+
self.optimizer._opt_called = True
27+
2328
# Create an optimizer container
2429
self.optimizer_container = MagicMock(spec=OptimizersContainer)
2530
self.optimizer_container.__iter__.return_value = iter([self.optimizer])

0 commit comments

Comments
 (0)