entrenar 0.7.11

Training & Optimization library with autograd, LoRA, quantization, and model merging
Documentation 
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
30
31
32
33
34
35
36
37
38
39
40
41
42

//! Callback system for training events
//!
//! Provides extensible hooks for training loop events:
//! - `on_train_begin` / `on_train_end`
//! - `on_epoch_begin` / `on_epoch_end`
//! - `on_step_begin` / `on_step_end`
//!
//! # Example
//!
//! ```rust
//! use entrenar::train::callback::{TrainerCallback, CallbackContext, CallbackAction};
//!
//! struct PrintCallback;
//!
//! impl TrainerCallback for PrintCallback {
//!     fn on_epoch_end(&mut self, ctx: &CallbackContext) -> CallbackAction {
//!         println!("Epoch {} finished with loss {:.4}", ctx.epoch, ctx.loss);
//!         CallbackAction::Continue
//!     }
//! }
//! ```

#![allow(clippy::field_reassign_with_default)]

mod checkpoint;
mod early_stopping;
mod explainability;
mod manager;
mod monitor;
mod progress;
mod scheduler;
mod traits;

// Re-export all public types
pub use checkpoint::CheckpointCallback;
pub use early_stopping::EarlyStopping;
pub use explainability::{ExplainMethod, ExplainabilityCallback, FeatureImportanceResult};
pub use manager::CallbackManager;
pub use monitor::MonitorCallback;
pub use progress::ProgressCallback;
pub use scheduler::LRSchedulerCallback;
pub use traits::{CallbackAction, CallbackContext, TrainerCallback};