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
Joshua Ferguson 706e0ebce2
Parser rewrite (#1296) 
* Running into issues with identity nodes

* Vec<RefCell<Node>> seems to work for this

* back to passing tests

* Reworked IO into separate struct

* working towards exploiting topological ordering and more informative ident errors

* the passing of an initializer to coalesce is temporary

* cleaning up dead code

* handled unsqueeze

* reworked node initialization and dim inference

* mainly cleanup

* changed how io use is tracked, moved unsqueeze remapping out of dim inference

* `cargo xtask run-checks all` now passes

* added a fixme and a few doc strings

* removing println and dead code

* spaces in doc strings

* altered top sort to work on node proto, moved prior to node gen

* Update ir.rs

* Update from_onnx.rs

removed dead code

* updated doc string

* camalcased Onnx Graph Builder

* removed self import?
 		2024-02-24 10:51:58 -06:00
..
data [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00
onnx-tests [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00
pytorch-tests [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00
src Parser rewrite (#1296) 2024-02-24 10:51:58 -06:00
Cargo.toml [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00
DEVELOPMENT.md [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00
LICENSE-APACHE [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00
LICENSE-MIT [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05: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 Instance Normalization (#1321) 2024-02-23 23:31:43 -06:00
build.rs [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -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.