tensorlogic-train 0.1.0

Training loops, loss composition, and optimization schedules for TensorLogic
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

//! Training callbacks for monitoring and controlling training.

pub mod advanced;
pub mod checkpoint;
pub mod core;
pub mod early_stopping;
pub mod gradient;
pub mod histogram;
pub mod lr_finder;
pub mod profiling;

// Re-export core types
pub use core::{BatchCallback, Callback, CallbackList, EpochCallback, ValidationCallback};

// Re-export checkpoint types
pub use checkpoint::{CheckpointCallback, CheckpointCompression, TrainingCheckpoint};

// Re-export early stopping types
pub use early_stopping::{EarlyStoppingCallback, ReduceLrOnPlateauCallback};

// Re-export lr_finder types
pub use lr_finder::LearningRateFinder;

// Re-export gradient types
pub use gradient::{
    GradientAccumulationCallback, GradientAccumulationStats, GradientMonitor,
    GradientScalingStrategy, GradientSummary,
};

// Re-export histogram types
pub use histogram::{HistogramCallback, HistogramStats};

// Re-export profiling types
pub use profiling::{ProfilingCallback, ProfilingStats};

// Re-export advanced types
pub use advanced::{ModelEMACallback, SWACallback};