mlr_callback_set.tb function

TensorBoard Logging Callback

Logs training loss, training measures, and validation measures as events. To view them, use TensorBoard with tensorflow::tensorboard() (requires tensorflow) or the CLI.

Details

Logs events at most every epoch.

See Also

Other Callback: TorchCallback, as_torch_callback(), as_torch_callbacks(), callback_set(), mlr3torch_callbacks, mlr_callback_set, mlr_callback_set.checkpoint, mlr_callback_set.progress, mlr_callback_set.unfreeze, mlr_context_torch, t_clbk(), torch_callback()

Super class

mlr3torch::CallbackSet -> CallbackSetTB

Methods

Public methods

• list("CallbackSetTB$new()")

• list("CallbackSetTB$on_epoch_end()")

• list("CallbackSetTB$clone()")

Method new()

Creates a new instance of this R6 class.

Usage

CallbackSetTB$new(path, log_train_loss)

Arguments

• path: (character(1))

   The path to a folder where the events are logged. Point TensorBoard to this folder to view them.
  

• log_train_loss: (logical(1))

   Whether we log the training loss.
  

Method on_epoch_end()

Logs the training loss, training measures, and validation measures as TensorBoard events.

Usage

CallbackSetTB$on_epoch_end()

Method clone()

The objects of this class are cloneable with this method.

Usage

CallbackSetTB$clone(deep = FALSE)

Arguments

• deep: Whether to make a deep clone.