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 76cb58d commit 6d8fff3Copy full SHA for 6d8fff3
tests/trainer/optimization/test_optimizers.py
@@ -582,6 +582,7 @@ def configure_optimizers(self):
582
@RunIf(min_gpus=2, special=True)
583
def test_optimizer_state_on_device(tmpdir):
584
""" Test that optimizers that create state initially at instantiation still end up with the state on the GPU. """
585
+
586
class TestModel(BoringModel):
587
588
def configure_optimizers(self):
0 commit comments