pub trait Callback<F: Float + Debug + ScalarOperand> {
// Required method
fn on_event(
&mut self,
timing: CallbackTiming,
context: &mut CallbackContext<'_, F>,
) -> Result<()>;
}
Expand description
Trait for training callbacks
Required Methods§
Sourcefn on_event(
&mut self,
timing: CallbackTiming,
context: &mut CallbackContext<'_, F>,
) -> Result<()>
fn on_event( &mut self, timing: CallbackTiming, context: &mut CallbackContext<'_, F>, ) -> Result<()>
Called during training at specific points