Skip to main content

entrenar/prune/data_loader/
mod.rs

1//! Calibration data loader for pruning
2//!
3//! Provides data loading utilities for collecting activation statistics
4//! during calibration for pruning methods like Wanda and SparseGPT.
5//!
6//! # Example
7//!
8//! ```ignore
9//! use entrenar::prune::{CalibrationDataLoader, CalibrationDataConfig};
10//!
11//! let config = CalibrationDataConfig::new()
12//!     .with_num_samples(128)
13//!     .with_batch_size(4);
14//!
15//! let loader = CalibrationDataLoader::new(config);
16//! for batch in loader.iter() {
17//!     // Process batch for calibration
18//! }
19//! ```
20
21mod config;
22mod iter;
23mod loader;
24
25#[cfg(test)]
26mod tests;
27
28pub use config::CalibrationDataConfig;
29pub use loader::CalibrationDataLoader;
30
31// CalibrationDataIter is publicly exported for users who need to reference
32// the iterator type returned by CalibrationDataLoader::iter().
33#[allow(unused_imports)]
34pub use iter::CalibrationDataIter;