Struct GradientCheckpoint 

pub struct GradientCheckpoint {
    pub id: String,
    pub saved_tensors: HashMap<String, Vec<f64>>,
    pub memory_saved: usize,
}

Gradient checkpointing utility.

Fields

id: String

Checkpoint identifier

saved_tensors: HashMap<String, Vec<f64>>

Saved tensors for recomputation

memory_saved: usize

Memory saved by checkpointing (bytes)

Implementations

impl GradientCheckpoint

pub fn new(id: String) -> Self

Create a new gradient checkpoint.

pub fn save_tensor(&mut self, name: String, data: Vec<f64>)

Save a tensor for recomputation.

pub fn memory_saved_mb(&self) -> f64

Get memory saved by this checkpoint.

Trait Implementations

impl Clone for GradientCheckpoint

fn clone(&self) -> GradientCheckpoint

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GradientCheckpoint

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

Formats the value using the given formatter.