Crate burn_import

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

burnonnx

The module for generating the burn code.

onnxonnx

The onnx module.

pytorchpytorch

The PyTorch module for recorder.

Functions

format_tokens

Formats a token stream into a string.