mlr_callback_set.tb {mlr3torch} | R Documentation |
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
Inherited methods
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()