Struct GradientCheckpoint 

pub struct GradientCheckpoint {
    pub graph: ComputationGraphStore,
    pub step: u64,
    pub loss_cid: Option<String>,
    pub optimizer_state: HashMap<String, Vec<u8>>,
    pub timestamp: u64,
}

A serializable snapshot of the gradient computation state.

Contains the full ComputationGraphStore plus training metadata so that a training run can be interrupted and resumed without recomputing the forward pass from scratch.

Fields

graph: ComputationGraphStore

The computation graph at the time of checkpointing

step: u64

Optimizer step counter

loss_cid: Option<String>

CID of the loss tensor (None before the first loss is computed)

optimizer_state: HashMap<String, Vec<u8>>

Raw optimizer state blobs, keyed by parameter name

timestamp: u64

Unix timestamp (seconds) when this checkpoint was created

Implementations

impl GradientCheckpoint

pub fn new(graph: ComputationGraphStore, step: u64) -> Self

Create a new checkpoint at the given step.

pub fn with_loss_cid(self, cid: impl Into<String>) -> Self

Attach the CID of the current loss tensor.

pub fn set_optimizer_state(&mut self, param: impl Into<String>, state: Vec<u8>)

Store a raw optimizer state blob for the given parameter.

pub fn save(&self, path: &Path) -> Result<(), GradientError>

Persist this checkpoint to a file at path.

The file is written as JSON so it is human-inspectable and version-control friendly.

pub fn load(path: &Path) -> Result<Self, GradientError>

Load a checkpoint from the file at path.

Trait Implementations

impl Debug for GradientCheckpoint

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for GradientCheckpoint

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for GradientCheckpoint

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.