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 2feb55f commit 25935feCopy full SHA for 25935fe
tests/tests_pytorch/utilities/test_meta.py
@@ -33,7 +33,7 @@ def __init__(self, num_layers: int):
33
self.layer = nn.Sequential(*[nn.Linear(1, 1) for _ in range(self.hparams.num_layers)])
34
35
36
-@RunIf(min_torch="1.10.0", standalone=True)
+@RunIf(min_torch="1.10.0")
37
def test_init_meta_context():
38
39
with init_meta_context():
@@ -72,7 +72,7 @@ def test_init_meta_context():
72
assert m.weight.device.type == "cpu"
73
74
75
76
def test_materialize_module_recursive_child():
77
"""Test materialize_module doesn't set a child recursively to a model instantiated within init_meta_context."""
78
0 commit comments