entrenar 0.7.12

Training & Optimization library with autograd, LoRA, quantization, and model merging
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

//! Distillation Trainer Orchestrator
//!
//! High-level API for knowledge distillation training.
//!
//! # Example
//!
//! ```ignore
//! use entrenar::hf_pipeline::{DistillationTrainer, TrainerConfig};
//!
//! let trainer = DistillationTrainer::new(TrainerConfig {
//!     teacher_model: "microsoft/codebert-base".into(),
//!     student_model: "distilbert-base-uncased".into(),
//!     ..Default::default()
//! });
//!
//! trainer.train(&dataset)?;
//! ```

mod config;
mod distillation_trainer;
mod state;

#[cfg(test)]
mod tests;

// Re-export all public types
pub use config::TrainerConfig;
pub use distillation_trainer::DistillationTrainer;
pub use state::TrainingState;