Language

unilab.training.experiment.ExperimentTracker¶

class unilab.training.experiment.ExperimentTracker[source]¶

Bases: object

Tracks experiment metadata locally and optionally in Weights & Biases.

Parameters:

root_dir (str | Path)
log_dir (str | Path)
algo_name (str)
task_name (str)
sim_backend (str)
training_cfg (Any)
full_cfg (Any)
device (str | None)
collector_device (str | None)
seed_info (Any | None)

Methods

`__init__`(*, root_dir, log_dir, algo_name, ...)
`finish`()
`log_video`(video_path[, key])
`start`()
`update_summary`([summary])

Attributes

`run`
`run_url`
`wandb_settings`

__init__(*, root_dir, log_dir, algo_name, task_name, sim_backend, training_cfg, full_cfg, device=None, collector_device=None, seed_info=None)[source]¶

Parameters:

root_dir (str | Path)
log_dir (str | Path)
algo_name (str)
task_name (str)
sim_backend (str)
training_cfg (Any)
full_cfg (Any)
device (str | None)
collector_device (str | None)
seed_info (Any | None)

property run: Any | None¶

property run_url: str | None¶

property wandb_settings: dict[str, Any]¶

start()[source]¶

Return type:: None

update_summary(summary=None)[source]¶

Parameters:: summary (dict[str, Any] | None)
Return type:: None

log_video(video_path, key='media/play_video')[source]¶

Parameters:

video_path (str | Path | None)
key (str)

Return type:

None

finish()[source]¶

Return type:: None