Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.
And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.
Checkpoint loading patch fails when using a pre-trained featurizer within a Lightning module.
See original GitHub issue🐛 Bug
I am getting the error: ModuleNotFoundError: No module named 'pytorch_lightning.utilities.argparse_utils' coming from
migration.py when loading checkpoints for models which leverage pre-trained models also trained via Pytorch Lightning.
Essentially, the context manager is called recursively, since the model relies on features generated from a pre-trained model. This model also requires a call to load_from_checkpoint(). As a result, the exit function of the context manager gets called twice in a row, with the error occurring on the second call.
Expected behavior
I would expect checkpoint loading to not raise en error because of a pre-trained featurizer model.
Environment
pytorch-lightning == 1.6.1
Issue Analytics
- State:
- Created a year ago
- Comments:6 (3 by maintainers)
Top Results From Across the Web
Top Related Medium Post
Top Related StackOverflow Question
Troubleshoot Live Code
Top Related Reddit Thread
Top Related Hackernoon Post
Top Related Tweet
Top Related Dev.to Post
Top Related Hashnode Post
Hi @carmocca, sorry for not coming back to this. I haven’t been able to reproduce this in a simple setting and the original code causing the issue has been simplified/resolved as well. I think there is still a scenario in which an issue might arise but it likely involves a convoluted example that can be more easily resolved outside of PL. So I think the issue can be closed for now.
Hi @jdavid04. Is this still an issue? Are you able to finish your PR?