burn-onnx 0.21.0-pre.3

Library for importing ONNX models into the Burn framework
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

#![warn(missing_docs)]
#![cfg_attr(docsrs, feature(doc_cfg))]

//! `burn-onnx` is a crate designed to simplify the process of importing models trained in other
//! machine learning frameworks into the Burn framework via the ONNX format. It generates the Rust
//! source code that aligns the imported model with Burn and converts tensor data into a compatible
//! format.

#[macro_use]
extern crate derive_new;

mod logger;

/// The module for generating the burn code.
pub mod burn;

mod formatter;
mod model_gen;

pub use formatter::*;
pub use model_gen::*;