Crate burn_import
source ·Expand description
burn-import is a crate designed to simplify the process of importing models trained in other
machine learning frameworks into the Burn framework. This tool generates a Rust source file that
aligns the imported model with Burn’s model and converts tensor data into a format compatible with
Burn.
Modules§
- The module for generating the burn code.
- The onnx module.
- The PyTorch module for recorder.
Functions§
- Formats a token stream into a string.