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 31f1607 commit 0c73b63Copy full SHA for 0c73b63
tests/tests_pytorch/trainer/connectors/test_checkpoint_connector.py
@@ -172,6 +172,8 @@ def test_loops_restore(tmpdir):
172
ckpt_path = str(tmpdir / "last.ckpt")
173
174
trainer = Trainer(**trainer_args)
175
+ trainer.strategy.connect(model)
176
+
177
for fn in TrainerFn:
178
if fn != TrainerFn.TUNING:
179
trainer_fn = getattr(trainer, f"{fn}_loop")
0 commit comments