Skip to main content

entrenar/hf_pipeline/trainer/
mod.rs

1//! Distillation Trainer Orchestrator
2//!
3//! High-level API for knowledge distillation training.
4//!
5//! # Example
6//!
7//! ```ignore
8//! use entrenar::hf_pipeline::{DistillationTrainer, TrainerConfig};
9//!
10//! let trainer = DistillationTrainer::new(TrainerConfig {
11//!     teacher_model: "microsoft/codebert-base".into(),
12//!     student_model: "distilbert-base-uncased".into(),
13//!     ..Default::default()
14//! });
15//!
16//! trainer.train(&dataset)?;
17//! ```
18
19mod config;
20mod distillation_trainer;
21mod state;
22
23#[cfg(test)]
24mod tests;
25
26// Re-export all public types
27pub use config::TrainerConfig;
28pub use distillation_trainer::DistillationTrainer;
29pub use state::TrainingState;