noether.core.trackers.tensorboard

Attributes

TENSORBOARD_IMPORT_ERROR

Classes

TensorboardTrackerSchema	Schema for TensorboardTracker configuration.
TensorboardTracker	TensorBoard tracker for logging metrics and configuration.

Module Contents

noether.core.trackers.tensorboard.TENSORBOARD_IMPORT_ERROR = None

class noether.core.trackers.tensorboard.TensorboardTrackerSchema(/, **data)

Bases: noether.core.trackers.BaseTrackerConfig

Schema for TensorboardTracker configuration.

Create a new model by parsing and validating input data from keyword arguments.

Raises [ValidationError][pydantic_core.ValidationError] if the input data cannot be validated to form a valid model.

self is explicitly positional-only to allow self as a field name.

Parameters:

data (Any)

log_dir: str = None

The base directory where TensorBoard event files will be stored.

flush_secs: int = None

How often, in seconds, to flush the pending events to disk.

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

Bases: noether.core.trackers.BaseTracker

TensorBoard tracker for logging metrics and configuration.

Initialize the TensorboardTracker.

Parameters:

• tracker_config (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