We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6b11290 commit 8004fefCopy full SHA for 8004fef
tests/unit_tests/test_lr_scheduler.py
@@ -20,6 +20,11 @@ def setUp(self):
20
self.model = torch.nn.Linear(10, 10)
21
# Create an optimizer
22
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
28
# Create an optimizer container
29
self.optimizer_container = MagicMock(spec=OptimizersContainer)
30
self.optimizer_container.__iter__.return_value = iter([self.optimizer])
0 commit comments