Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Use Burn's PyTorchFileRecoder to Directly Load PyTorch Model #25

Open
antimora opened this issue May 2, 2024 · 0 comments
Open

Use Burn's PyTorchFileRecoder to Directly Load PyTorch Model #25

antimora opened this issue May 2, 2024 · 0 comments

Comments

@antimora
Copy link

antimora commented May 2, 2024

Currently, the process of loading a PyTorch model into Burn requires an extra step of converting the model to Burn's format. An improvement that could streamline this process is to utilize Burn's PyTorchFileRecoder to load the PyTorch model directly.

The PyTorchFileRecoder is a convenient tool provided by Burn that allows direct loading of a PyTorch model file (.pt) and saves it into Burn's format. This eliminates the need for a separate conversion step and simplifies the overall workflow.

For more detailed information and examples, please refer to the Burn documentation on importing PyTorch models: https://burn.dev/book/import/pytorch-model.html

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant