Enum CheckpointPolicy 

pub enum CheckpointPolicy {
    EveryNLayers(usize),
    SpecificLayers(Vec<usize>),
    Automatic {
        memory_budget: usize,
        avg_activation_size: usize,
    },
    Custom,
    None,
}

Policy for selecting which activations to checkpoint

Variants

EveryNLayers(usize)

Checkpoint every N layers (most common for transformers)

SpecificLayers(Vec<usize>)

Checkpoint at specific layer indices

Automatic

Automatic policy based on memory constraints

Fields

memory_budget: usize

Target memory budget in bytes

avg_activation_size: usize

Estimated activation size per layer in bytes

Custom

Custom policy via user-defined function

None

No checkpointing (store all activations)

Trait Implementations

impl Clone for CheckpointPolicy

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for CheckpointPolicy

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

Formats the value using the given formatter.

impl Default for CheckpointPolicy

fn default() -> Self

Returns the “default value” for a type.

impl PartialEq for CheckpointPolicy

fn eq(&self, other: &Self) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl StructuralPartialEq for CheckpointPolicy