Struct GradientCheckpointConfig 

pub struct GradientCheckpointConfig {
    pub enabled: bool,
    pub strategy: CheckpointStrategy,
    pub checkpoint_layers: Vec<String>,
    pub memory_threshold: Option<usize>,
}

Gradient checkpointing configuration.

Gradient checkpointing reduces memory usage by recomputing activations during the backward pass instead of storing them. This trades compute for memory.

Fields

enabled: bool

Whether gradient checkpointing is enabled.

strategy: CheckpointStrategy

Checkpointing strategy.

checkpoint_layers: Vec<String>

Layers to checkpoint (by name pattern).

memory_threshold: Option<usize>

Memory threshold to trigger checkpointing (bytes).

Implementations

impl GradientCheckpointConfig

pub fn new() -> Self

Create a new gradient checkpoint config.

pub fn enabled(self) -> Self

Enable gradient checkpointing.

pub fn with_strategy(self, strategy: CheckpointStrategy) -> Self

Set checkpointing strategy.

pub fn with_layers(self, layers: Vec<String>) -> Self

Set layers to checkpoint.

pub fn with_memory_threshold(self, threshold: usize) -> Self

Set memory threshold for automatic checkpointing.

Trait Implementations

impl Clone for GradientCheckpointConfig

fn clone(&self) -> GradientCheckpointConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GradientCheckpointConfig

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

Formats the value using the given formatter.

impl Default for GradientCheckpointConfig

fn default() -> Self

Returns the “default value” for a type.