Struct Checkpoint 

pub struct Checkpoint {
    pub tensors: HashMap<String, Tensor>,
    pub metadata: CheckpointMetadata,
}

Model checkpoint

Fields

tensors: HashMap<String, Tensor>

Model parameters

metadata: CheckpointMetadata

Metadata

Implementations

impl Checkpoint

pub fn new( tensors: HashMap<String, Tensor>, metadata: CheckpointMetadata, ) -> Self

Create new checkpoint

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

Save checkpoint to file

Arguments

• path - Path to save checkpoint

Returns

Result indicating success or error

pub fn load<P: AsRef<Path>>(path: P, device: &Device) -> Result<Self>

Load checkpoint from file

Arguments

• path - Path to checkpoint file
• device - Device to load tensors on

Returns

Loaded checkpoint

pub fn load_weights<P: AsRef<Path>>( path: P, device: &Device, ) -> Result<HashMap<String, Tensor>>

Load only model weights from checkpoint

Arguments

• path - Path to checkpoint file
• device - Device to load tensors on

Returns

HashMap of parameter name to tensor