noether.core.trackers.wandb_tracker

Attributes

WANDB_IMPORT_ERROR

Classes

WandBTracker

Weights and Biases tracker.

Module Contents

noether.core.trackers.wandb_tracker.WANDB_IMPORT_ERROR = None

class noether.core.trackers.wandb_tracker.WandBTracker(tracker_config, **kwargs)

Bases: noether.core.trackers.base.BaseTracker

Weights and Biases tracker.

Example config:

- kind: noether.core.trackers.WandBTracker
  mode: online
  entity: my_entity
  project: my_project

Parameters:

• tracker_config (noether.core.schemas.trackers.WandBTrackerSchema) – Configuration for the WandBTracker. See WandBTrackerSchema for available options including mode, entity, and project.

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

MODES = ['disabled', 'online', 'offline']

mode

entity

project