Struct CheckpointConfig 

pub struct CheckpointConfig {
    pub checkpoint_every: usize,
    pub enabled: bool,
}

Gradient checkpointing configuration.

Controls how activations are stored during forward pass. Higher checkpoint_every values reduce memory but increase compute.

Fields

checkpoint_every: usize

Checkpoint every N layers

enabled: bool

Enable checkpointing

Implementations

impl CheckpointConfig

pub fn new(checkpoint_every: usize, enabled: bool) -> CheckpointConfig

Create a new checkpoint config.

pub fn memory_reduction_factor(&self, num_layers: usize) -> f64

Calculate expected memory reduction factor.

Returns a value between 0 and 1, where lower is better.

Trait Implementations

impl Clone for CheckpointConfig

fn clone(&self) -> CheckpointConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for CheckpointConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl Default for CheckpointConfig

fn default() -> CheckpointConfig

Returns the “default value” for a type.