tsai_train 0.1.2

Training loop, callbacks, metrics, and checkpointing for tsai-rs
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

//! Error types for training.

use thiserror::Error;

/// Result type alias for training operations.
pub type Result<T> = std::result::Result<T, TrainError>;

/// Errors that can occur during training.
#[derive(Error, Debug)]
pub enum TrainError {
    /// Model forward pass failed.
    #[error("Forward pass failed: {0}")]
    ForwardError(String),

    /// Backward pass failed.
    #[error("Backward pass failed: {0}")]
    BackwardError(String),

    /// Optimizer step failed.
    #[error("Optimizer step failed: {0}")]
    OptimizerError(String),

    /// Invalid learning rate.
    #[error("Invalid learning rate: {0}")]
    InvalidLearningRate(String),

    /// Checkpoint error.
    #[error("Checkpoint error: {0}")]
    CheckpointError(String),

    /// Callback error.
    #[error("Callback error: {0}")]
    CallbackError(String),

    /// Data error.
    #[error("Data error: {0}")]
    DataError(#[from] tsai_data::DataError),

    /// Core error.
    #[error("Core error: {0}")]
    CoreError(#[from] tsai_core::CoreError),

    /// I/O error.
    #[error("I/O error: {0}")]
    IoError(#[from] std::io::Error),

    /// Serialization error.
    #[error("Serialization error: {0}")]
    SerializationError(String),

    /// Training was interrupted.
    #[error("Training interrupted: {0}")]
    Interrupted(String),

    /// Other error.
    #[error("{0}")]
    Other(String),
}