Struct ModelCheckpoint 

pub struct ModelCheckpoint<T: Float> {
    pub epoch: usize,
    pub step: usize,
    pub model_state: StateDict<T>,
    pub optimizer_state: HashMap<String, Vec<u8>>,
    pub scheduler_state: HashMap<String, Vec<u8>>,
    pub metrics: HashMap<String, f64>,
    pub timestamp: u64,
}

Model checkpoint system モデルチェックポイントシステム

Fields

epoch: usize

step: usize

model_state: StateDict<T>

optimizer_state: HashMap<String, Vec<u8>>

scheduler_state: HashMap<String, Vec<u8>>

metrics: HashMap<String, f64>

timestamp: u64

Implementations

impl<T: Float + 'static> ModelCheckpoint<T>

pub fn new(epoch: usize, step: usize, model_state: StateDict<T>) -> Self

Create new model checkpoint 新しいモデルチェックポイントを作成

pub fn add_optimizer_state(&mut self, name: String, state: Vec<u8>)

Add optimizer state オプティマイザー状態を追加

pub fn add_metric(&mut self, name: String, value: f64)

Add training metrics トレーニングメトリクスを追加

pub fn save_checkpoint<P: AsRef<Path>>( &self, path: P, ) -> SerializationResult<()>

Save checkpoint to file チェックポイントをファイルに保存

pub fn load_checkpoint<P: AsRef<Path>>(path: P) -> SerializationResult<Self>

Load checkpoint from file ファイルからチェックポイントを読み込み

Trait Implementations

impl<T: Clone + Float> Clone for ModelCheckpoint<T>

fn clone(&self) -> ModelCheckpoint<T>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<T: Debug + Float> Debug for ModelCheckpoint<T>

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

Formats the value using the given formatter.

impl<T: Float + 'static> Loadable for ModelCheckpoint<T>

fn load_binary(data: &[u8]) -> SerializationResult<Self>

Load object from binary format バイナリ形式からオブジェクトを読み込み

fn expected_type_id() -> &'static str

Get expected type identifier 期待されるタイプ識別子を取得

fn validate_version(version: &str) -> SerializationResult<()>

Validate version compatibility バージョン互換性を検証

impl<T: Float + 'static> Saveable for ModelCheckpoint<T>

fn save_binary(&self) -> SerializationResult<Vec<u8>>

Save object to binary format オブジェクトをバイナリ形式で保存

fn type_id(&self) -> &'static str

Get object type identifier オブジェクトタイプ識別子を取得

fn metadata(&self) -> HashMap<String, String>

Get metadata for object オブジェクトのメタデータを取得

fn version(&self) -> String

Get version information バージョン情報を取得