scirs2-neural 0.2.0

Neural network building blocks module for SciRS2 (scirs2-neural) - Minimal Version
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

//! Gradient checkpointing utilities

/// Configuration for gradient checkpointing
#[derive(Debug, Clone)]
pub struct GradientCheckpointingConfig {
    /// Whether to enable gradient checkpointing
    pub enabled: bool,
    /// Checkpointing frequency
    pub checkpoint_frequency: usize,
}

impl Default for GradientCheckpointingConfig {
    fn default() -> Self {
        Self {
            enabled: false,
            checkpoint_frequency: 1,
        }
    }
}