noether.core.trackers.tensorboard

Attributes

TENSORBOARD_IMPORT_ERROR

Classes

TensorboardTracker

TensorBoard tracker for logging metrics and configuration.

Module Contents

noether.core.trackers.tensorboard.TENSORBOARD_IMPORT_ERROR = None

class noether.core.trackers.tensorboard.TensorboardTracker(tracker_config, **kwargs)

Bases: noether.core.trackers.base.BaseTracker

TensorBoard tracker for logging metrics and configuration.

Initialize the TensorboardTracker.

Parameters:

• tracker_config (noether.core.schemas.trackers.TensorboardTrackerSchema) – Configuration for TensorBoard. See TensorboardTrackerSchema for available options (typically including base log_dir).

• **kwargs – Additional keyword arguments passed to the parent class.

config

writer: torch.utils.tensorboard.SummaryWriter | None = None