mirror of https://github.com/tracel-ai/burn.git
9b9b03c959
* add random normal like python code to generate onnx model * add random normal like node * modify onnx burn to add new op * add test on test onnx * revert commentouts * fix review points to respond to dynamically shape * add onnx op of random uniform like * fix formats |
||
---|---|---|
.. | ||
onnx-tests | ||
pytorch-tests | ||
src | ||
Cargo.toml | ||
DEVELOPMENT.md | ||
LICENSE-APACHE | ||
LICENSE-MIT | ||
README.md | ||
SUPPORTED-ONNX-OPS.md |
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:
-
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.