drytorch.trackers.hydra

Module containing the HydraLink tracker.

Classes

HydraLink([par_dir, copy_hydra, hydra_dir])

Link current Hydra metadata to the experiment.

class HydraLink(par_dir: Path | None = None, copy_hydra: bool = True, hydra_dir: Path | None = None)

Bases: Dumper

Link current Hydra metadata to the experiment.

hydra_dir

the directory where hydra saves the run.

Type:

pathlib.Path

Initialize.

Parameters:

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

• copy_hydra (bool) – if True, copy the hydra folder content at the end of the experiment’s scope, replacing the link folder.

• hydra_dir (Path) – the directory where hydra saves the run.

Raises:

TrackerError – if hydra has not started (hydra_dir does not exist).

clean_up() → None

Remove experimental data from the tracker.

Return type:

None

notify(event: Event) → None

notify(event: StartExperimentEvent) → None

Notify the tracker of an event.

Parameters:

event (Event) – the event to notify about.

Return type:

None