9 releases (breaking)
0.13.0 | Apr 12, 2024 |
---|---|
0.12.1 | Feb 1, 2024 |
0.12.0 | Jan 31, 2024 |
0.11.1 | Dec 4, 2023 |
0.7.0 | May 6, 2023 |
#812 in Machine learning
293 downloads per month
Used in fibertools-rs
340KB
7.5K
SLoC
Importing Models
The Burn project supports the import of models from various frameworks, emphasizing efficiency and compatibility. Currently, it handles two primary model formats:
-
ONNX: Facilitates direct import, ensuring the model's performance and structure are maintained.
-
PyTorch: Enables the loading of PyTorch model weights into Burn’s native model architecture, ensuring seamless integration.
Contribution
Interested in contributing to burn-import
? Check out our development guide for
more information.
lib.rs
:
burn-import
is a crate designed to simplify the process of importing models trained in other
machine learning frameworks into the Burn framework. This tool generates a Rust source file that
aligns the imported model with Burn's model and converts tensor data into a format compatible with
Burn.
Dependencies
~38–76MB
~1.5M SLoC