Struct CheckpointConfig 

pub struct CheckpointConfig {
    pub checkpoint_every_n_layers: usize,
    pub max_checkpoints: usize,
    pub use_mixed_precision: bool,
}

Configuration for activation checkpointing.

Fields

checkpoint_every_n_layers: usize

Save an activation checkpoint every N layers.

For example, 3 means layers 0, 3, 6, 9, … will have their activations stored.

max_checkpoints: usize

Maximum number of checkpoint slots.

Once this limit is reached, older checkpoints may be evicted.

use_mixed_precision: bool

Whether to use reduced precision (f16-like truncation) for stored activations to further save memory.

When enabled, stored activations are quantised to half-precision (simulated by rounding to 3 decimal places) and restored on retrieval.

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

Formats the value using the given formatter.

impl Default for CheckpointConfig

fn default() -> Self

Returns the “default value” for a type.