mlr_callback_set.tb: TensorBoard Logging Callback

In mlr3torch: Deep Learning with 'mlr3'

TensorBoard Logging Callback

Description

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.

Super class

mlr3torch::CallbackSet -> CallbackSetTB

Methods

Public methods

• CallbackSetTB$new()

• CallbackSetTB$on_epoch_end()

• CallbackSetTB$clone()

Inherited methods

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

---

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.

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()

mlr3torch documentation built on April 4, 2025, 3:03 a.m.