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.

Resuming

This callback keeps no state of its own so it can trivially be resumed.

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

CallbackSet -> CallbackSetTB

Methods

Public methods

• CallbackSetTB$new()

• CallbackSetTB$on_epoch_end()

• CallbackSetTB$clone()

Inherited methods

• CallbackSet$load_state_dict()
• CallbackSet$print()
• CallbackSet$state_dict()

---

CallbackSetTB$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. The folder must be new, empty, or one that this callback already logged into, so that a resumed run can continue the log of the run it continues.

log_train_loss

(logical(1))
Whether we log the training loss.

---

CallbackSetTB$on_epoch_end()

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

Usage

CallbackSetTB$on_epoch_end()

---

CallbackSetTB$clone()

The objects of this class are cloneable with this method.

Usage

CallbackSetTB$clone(deep = FALSE)

Arguments

deep

Whether to make a deep clone.