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§