Open Issues Need Help
View All on GitHub AI/ML • Model Management
Loading model from .pth is not supported yet 28 days ago
AI Summary: The task is to fix a bug in the LitModels library where loading PyTorch models saved as '.pth' files is not supported. The issue is that the `save_model` function saves PyTorch models in '.pth' format, but the `load_model` function doesn't handle this format, resulting in a `NotImplementedError`. The solution involves extending the `load_model` function to support loading '.pth' files, likely by adding handling for the specific file extension and using appropriate PyTorch loading mechanisms.
Complexity:
3/5
bug help wanted
Save, load, host, and share AI model checkpoints without slowing down training. Host on Lightning AI or your own cloud with enterprise-grade access controls.
Python
#checkpoint#fine-tuning#model#sharing