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