Skip to main content

entrenar/yaml_mode/manifest/
mod.rs

1//! Training Manifest Schema
2//!
3//! Defines the complete YAML Mode Training manifest structure as specified in
4//! docs/specifications/yaml-mode-train.md
5//!
6//! This module is organized into submodules for better maintainability:
7//! - `core` - TrainingManifest root struct
8//! - `data` - DataConfig, DataSplit, PreprocessingStep, DataLoader, etc.
9//! - `model` - ModelConfig, ArchitectureConfig
10//! - `optimizer` - OptimizerConfig, ParamGroup
11//! - `scheduler` - SchedulerConfig, WarmupConfig
12//! - `training` - TrainingConfig, GradientConfig, MixedPrecisionConfig, etc.
13//! - `lora` - LoraConfig
14//! - `quantize` - QuantizeConfig, QatConfig, CalibrationConfig
15//! - `monitoring` - MonitoringConfig, DriftDetectionConfig, TerminalMonitor, etc.
16//! - `callback` - CallbackConfig, CallbackType
17//! - `output` - OutputConfig, ModelOutputConfig, MetricsOutputConfig, etc.
18//! - `extended` - Extended configurations for YAML Mode QA Epic
19
20pub mod callback;
21pub mod core;
22pub mod data;
23pub mod extended;
24pub mod lora;
25pub mod model;
26pub mod monitoring;
27pub mod optimizer;
28pub mod output;
29pub mod publish;
30pub mod quantize;
31pub mod scheduler;
32pub mod shorthand;
33pub mod training;
34
35// Re-export all public types for API compatibility
36pub use callback::{CallbackConfig, CallbackType};
37pub use core::TrainingManifest;
38pub use data::{DataConfig, DataLoader, DataSplit, PreprocessingStep};
39pub use extended::{
40    AuditConfig, BackpressureConfig, BenchmarkConfig, CitlConfig, DebugConfig, DistillModelRef,
41    DistillationConfig, GraphConfig, InspectConfig, PrivacyConfig, RagConfig, SessionConfig,
42    SigningConfig, StressConfig, VerificationConfig,
43};
44pub use lora::LoraConfig;
45pub use model::ModelConfig;
46pub use monitoring::{
47    AlertConfig, ChartConfig, DriftDetectionConfig, MonitoringConfig, SystemMonitorConfig,
48    TerminalMonitor, TrackingConfig,
49};
50pub use optimizer::OptimizerConfig;
51pub use output::{
52    MetricsOutputConfig, ModelOutputConfig, OutputConfig, RegistryConfig, ReportConfig,
53};
54pub use publish::PublishConfig;
55pub use quantize::QuantizeConfig;
56pub use scheduler::{SchedulerConfig, WarmupConfig};
57pub use training::{
58    CheckpointConfig, EarlyStoppingConfig, GradientConfig, MixedPrecisionConfig, TrainingConfig,
59};