burn_onnx/lib.rs
1#![warn(missing_docs)]
2#![cfg_attr(docsrs, feature(doc_cfg))]
3
4//! `burn-onnx` is a crate designed to simplify the process of importing models trained in other
5//! machine learning frameworks into the Burn framework via the ONNX format. It generates the Rust
6//! source code that aligns the imported model with Burn and converts tensor data into a compatible
7//! format.
8
9#[macro_use]
10extern crate derive_new;
11
12mod logger;
13
14/// The module for generating the burn code.
15pub mod burn;
16
17mod formatter;
18mod model_gen;
19
20pub use formatter::*;
21pub use model_gen::*;