Struct burn::train::checkpoint::FileCheckpointer

pub struct FileCheckpointer<FR> { /* private fields */ }

The file checkpointer.

Implementations

impl<FR> FileCheckpointer<FR>

pub fn new( recorder: FR, directory: impl AsRef<Path>, name: &str, ) -> FileCheckpointer<FR>

Creates a new file checkpointer.

Arguments

• recorder - The file recorder.
• directory - The directory to save the checkpoints.
• name - The name of the checkpoint.

Trait Implementations

impl<FR, R, B> Checkpointer<R, B> for FileCheckpointer<FR>

where R: Record<B>, FR: FileRecorder<B>, B: Backend,

fn save(&self, epoch: usize, record: R) -> Result<(), CheckpointerError>

Save the record.

fn restore( &self, epoch: usize, device: &<B as Backend>::Device, ) -> Result<R, CheckpointerError>

Restore the record.

fn delete(&self, epoch: usize) -> Result<(), CheckpointerError>

Delete the record at the given epoch if present.