Skip to content

Commit

Permalink
restore
Browse files Browse the repository at this point in the history
  • Loading branch information
awaelchli committed Jun 12, 2021
1 parent 0d58eb2 commit 6d453f6
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions tests/trainer/test_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -428,10 +428,7 @@ def test_model_checkpoint_only_weights(tmpdir):

# assert restoring train state fails
with pytest.raises(KeyError, match="checkpoint contains only the model"):
trainer.checkpoint_connector.resume_from_checkpoint(new_weights_path)
trainer.checkpoint_connector.resume_start()
trainer.checkpoint_connector.restore_training_state()
trainer.checkpoint_connector.resume_end()
trainer.checkpoint_connector.restore(checkpoint_path)


def test_model_freeze_unfreeze():
Expand Down

0 comments on commit 6d453f6

Please sign in to comment.