Struct ModelCheckpoint 

pub struct ModelCheckpoint {
    pub state: StateDict,
    pub meta: CheckpointMeta,
    pub hparams: HyperparamConfig,
}

A model checkpoint: state dict + metadata + hyperparameters.

Fields

state: StateDict

Model weights.

meta: CheckpointMeta

Checkpoint metadata.

hparams: HyperparamConfig

Hyperparameters used when this checkpoint was saved.

Implementations

impl ModelCheckpoint

pub fn new( state: StateDict, meta: CheckpointMeta, hparams: HyperparamConfig, ) -> Self

Create a new checkpoint.

pub fn to_bytes(&self) -> Vec<u8>

Serialise to a flat byte buffer.

Format: \[state_len: u64\][state_bytes]\[meta_text_len: u64\][meta_text_utf8]\[hp_json_len: u64\][hp_json_utf8]

pub fn byte_size(&self) -> usize

Byte size of the serialised checkpoint.

Trait Implementations

impl Clone for ModelCheckpoint

fn clone(&self) -> ModelCheckpoint

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ModelCheckpoint

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

Formats the value using the given formatter.