scirs2_neural/training/
gradient_checkpointing.rs

1//! Gradient checkpointing utilities
2
3/// Configuration for gradient checkpointing
4#[derive(Debug, Clone)]
5pub struct GradientCheckpointingConfig {
6    /// Whether to enable gradient checkpointing
7    pub enabled: bool,
8    /// Checkpointing frequency
9    pub checkpoint_frequency: usize,
10}
11
12impl Default for GradientCheckpointingConfig {
13    fn default() -> Self {
14        Self {
15            enabled: false,
16            checkpoint_frequency: 1,
17        }
18    }
19}