dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
burn/burn-import
History
Dilshod Tadjibaev 44266d5fd4
Make all struct CamelCase (#1316) 		2024-02-15 13:00:37 -06: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 Unsqueeze op (#1236) 2024-02-15 10:33:41 -05:00
pytorch-tests Make all struct CamelCase (#1316) 2024-02-15 13:00:37 -06:00
src Make all struct CamelCase (#1316) 2024-02-15 13:00:37 -06: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 Unsqueeze op (#1236) 2024-02-15 10:33:41 -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 Burn’s native model architecture, ensuring seamless integration.

Contribution

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