opentslm 0.1.0

Rust implementation of OpenTSLM using Burn, WGPU, and llama.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

//! Training infrastructure: curriculum trainer and metrics collection.
//!
//! # Sub-modules
//!
//! - [`curriculum`] — [`CurriculumTrainer`] orchestrates the five-stage
//!   progressive training loop.  Each stage loads its dataset, runs AdamW
//!   optimisation with cosine LR scheduling and early stopping, saves
//!   checkpoints, and writes predictions for the test set.
//!
//! - [`metrics`] — [`metrics::StageMetrics`] accumulates per-epoch numbers (loss,
//!   perplexity, accuracy, macro recall) and serialises them to CSV, four SVG
//!   plots, and a self-contained HTML dashboard.
//!
//! [`CurriculumTrainer`]: curriculum::CurriculumTrainer

pub mod curriculum;
pub mod metrics;