Struct PtCheckpoint 

pub struct PtCheckpoint { /* private fields */ }

Implementations

impl PtCheckpoint

pub fn load(path: impl AsRef<Path>, opts: LoadOptions) -> Result<Self>

pub fn from_metadata( metadata: CheckpointMetadata, source: ReconstructSource, ) -> Result<Self>

pub fn metadata(&self) -> &CheckpointMetadata

pub fn source_sha256(&self) -> &str

pub fn warnings(&self) -> &[String]

pub fn tensor_count(&self) -> usize

pub fn state_dict(&self) -> Result<BTreeMap<String, TensorArray>>

pub fn export( &self, out_dir: impl AsRef<Path>, opts: ExportOptions, ) -> Result<ExportResult>

Trait Implementations

impl Clone for PtCheckpoint

fn clone(&self) -> PtCheckpoint

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PtCheckpoint

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

Formats the value using the given formatter.