Expand description
burn-onnx is a crate designed to simplify the process of importing models trained in other
machine learning frameworks into the Burn framework via the ONNX format. It generates the Rust
source code that aligns the imported model with Burn and converts tensor data into a compatible
format.
Modules§
- burn
- The module for generating the burn code.
Structs§
- Model
Gen - Builder for generating Burn model code from ONNX files.
Enums§
- Load
Strategy - Controls how model weights are loaded at runtime.
Functions§
- format_
tokens - Formats a token stream into a string.