burn/burn-import
Guillaume Lagrange c603c68258
Fix chain pattern matching when multiple patterns are provided (#1273)
2024-02-07 16:51:48 -05:00
..
data Add support for loading PyTorch `.pt` (weights/states) files directly to model's record (#1085) 2024-01-25 10:20:09 -05:00
onnx-tests Elementwise pow op (#1133) 2024-01-24 09:46:57 -05:00
pytorch-tests Fix chain pattern matching when multiple patterns are provided (#1273) 2024-02-07 16:51:48 -05:00
src Regex fix pytorch (#1196) 2024-01-31 09:53:52 -05:00
Cargo.toml Bump Burn version to 0.13 (#1211) 2024-01-31 16:01:20 -05:00
DEVELOPMENT.md Update burn-import README (#727) 2023-08-30 08:20:53 -04:00
LICENSE-APACHE License fixes (#648) 2023-08-16 12:45:35 -04:00
LICENSE-MIT License fixes (#648) 2023-08-16 12:45:35 -04:00
README.md Update documentation and book sections on PyTorch import (#1180) 2024-01-29 17:55:20 -05:00
SUPPORTED-ONNX-OPS.md Elementwise pow op (#1133) 2024-01-24 09:46:57 -05:00
build.rs Add foundation for importing ONNX files (#297) 2023-04-15 10:44:50 -04:00

README.md

Importing Models

The Burn project supports the import of models from various frameworks, emphasizing efficiency and compatibility. Currently, it handles two primary model formats:

  1. ONNX: Facilitates direct import, ensuring the model's performance and structure are maintained.

  2. PyTorch: Enables the loading of PyTorch model weights into Burns native model architecture, ensuring seamless integration.

Contribution

Interested in contributing to burn-import? Check out our development guide for more information.