dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
burn/crates/burn-import
History
Dilshod Tadjibaev 297173124f
Add 1d and 2d modules for interpolate with scaling (also fix ONNX Resize op) (#2081) 
* Add interpolate module

* Update module.md

* Add interpolate 1d and 2d modules

* Consolidated InterpolateMode for 1d and 2d

* Remove CoordinateTransformationMode

* Add 1d tests for interpolate

* Refactor and fixes of ONNX Resize OP

* Fix clippy

* Fix docs

* Fix no_std
 		2024-07-31 12:08:26 -05:00
..
onnx-tests Add 1d and 2d modules for interpolate with scaling (also fix ONNX Resize op) (#2081) 2024-07-31 12:08:26 -05:00
pytorch-tests Print module - implement module display for remaining modules (part2) (#1933) 2024-06-28 08:37:40 -04:00
src Add 1d and 2d modules for interpolate with scaling (also fix ONNX Resize op) (#2081) 2024-07-31 12:08:26 -05:00
Cargo.toml Separating ONNX parsing from burn-import (#1921) 2024-07-02 15:17:44 -05:00
DEVELOPMENT.md Add subtract tensor from scalar for ONNX sub op (#1964) 2024-07-05 13:52:02 -05:00
LICENSE-APACHE Update licenses symlinks (#1613) 2024-04-12 14:43:58 -04:00
LICENSE-MIT Update licenses symlinks (#1613) 2024-04-12 14:43:58 -04:00
README.md [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00
SUPPORTED-ONNX-OPS.md Implement ONNX ConstantOfShape (#1815) 2024-07-08 15:11:59 -05: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.