Crate burn_import

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§

burnonnx
The module for generating the burn code.
onnxonnx
The onnx module.
pytorchpytorch
The PyTorch module for recorder.
safetensorssafetensors
The Safetensors module for recorder.

Functions§

format_tokens
Formats a token stream into a string.