training/lib.rs
1#![recursion_limit = "256"]
2
3pub mod dataset;
4pub mod util;
5
6pub use dataset::{collate, collate_from_burn_batch, CollatedBatch, DatasetConfig, RunSample};
7pub use models::{BigDet, BigDetConfig, TinyDet, TinyDetConfig};
8pub use util::{run_train, TrainArgs};
9/// Backend alias for training/eval (NdArray by default; WGPU if enabled).
10#[cfg(feature = "backend-wgpu")]
11pub type TrainBackend = burn_wgpu::Wgpu<f32>;
12#[cfg(not(feature = "backend-wgpu"))]
13pub type TrainBackend = burn_ndarray::NdArray<f32>;