OA LibraryReference

oa::ml::training::TrainingCallbackContext

Restricted mutable services available to one callback invocation.

StRust public API

Declaration

pub struct TrainingCallbackContext<'a> { /* private fields */ }

Public Rust declaration.

Public Methods

pub const fn snapshot(&self) -> TrainingSnapshot

pub fn optimizer(&mut self) -> &mut dyn Optimizer

pub fn checkpoint_optimizer(&mut self) -> Result<&mut dyn CheckpointOptimizer>

pub const fn config(&self) -> &ItTrainingConfig

pub fn exclude_wall_time(&mut self, duration: Duration)

Public Method Documentation

pub const fn snapshot(
	&self
) -> TrainingSnapshot

No public source comment is attached to this declaration.

pub fn optimizer(
	&mut self
) -> &mut dyn Optimizer

No public source comment is attached to this declaration.

pub fn checkpoint_optimizer(
	&mut self
) -> Result<&mut dyn CheckpointOptimizer>

No public source comment is attached to this declaration.

pub const fn config(
	&self
) -> &ItTrainingConfig

No public source comment is attached to this declaration.

pub fn exclude_wall_time(
	&mut self,
	duration: Duration
)

No public source comment is attached to this declaration.