Struct TensorBoardCallback 

pub struct TensorBoardCallback { /* private fields */ }

Training callback that logs scalar metrics to TensorBoard event files.

Implementations

impl TensorBoardCallback

pub fn new(log_dir: impl AsRef<Path>) -> Result<Self, ModelError>

Create a new callback writing events to log_dir.

pub fn global_step(&self) -> i64

Returns the current global step counter.

Trait Implementations

impl TrainingCallback for TensorBoardCallback

fn on_epoch_end( &mut self, _epoch: usize, metrics: &HashMap<String, f32>, ) -> bool

Called after each epoch. Returns true if training should stop.

fn on_batch_end(&mut self, _epoch: usize, _batch: usize, loss: f32)

Called after each batch within an epoch. Default implementation does nothing.