drytorch.trackers.wandb

Module containing a tracker calling Weights and Biases.

Classes

Wandb([par_dir, settings])

Tracker that wraps a run for the wandb library.

Exceptions

WandbWarning(*args)

Warning class for wandb.

class Wandb(par_dir: Path | None = None, settings: Settings = wandb.sdk.wandb_settings.Settings())

Bases: Dumper

Tracker that wraps a run for the wandb library.

Initialize.

Parameters:

• par_dir (Path | None) – the parent directory for the tracker data. Default uses the same of the current experiment.

• settings (Settings) – settings object from wandb containing all init arguments.

property run: Run

Active wandb run instance.

Raises:

AccessOutsideScopeError – if no run has been started yet.

clean_up() → None

Remove experimental data from the tracker.

Return type:

None

notify(event: Event) → None

notify(event: StartExperimentEvent) → None

notify(event: StopExperimentEvent) → None

notify(event: MetricEvent) → None

Notify the tracker of an event.

Parameters:

event (Event) – the event to notify about.

Return type:

None

exception WandbWarning(*args: Any)

Bases: DryTorchWarning

Warning class for wandb.

Initialize.

Parameters:

*args (Any) – arguments to be formatted into the message template.

Return type:

None