nemo_gym.exporters.wandb

Module Contents

Name	Description
`WandbExporter`	Weights & Biases backend.

class nemo_gym.exporters.wandb.WandbExporter(
    global_config_dict: omegaconf.DictConfig
)

Bases: BaseExporter

Weights & Biases backend.

Configured by wandb_project, wandb_name and wandb_api_key in the global config.

ROLLOUTS_TABLE_KEY

str = 'Rollouts'

config

= WANDBConfig.model_validate(global_config_dict)

name

str = 'wandb'

run

Optional[Run] = None

nemo_gym.exporters.wandb.WandbExporter._active_run() -> wandb.Run

nemo_gym.exporters.wandb.WandbExporter._log_config(
    config_dict: omegaconf.DictConfig
) -> None

nemo_gym.exporters.wandb.WandbExporter._log_metrics(
    metrics: dict[str, typing.Any],
    step: typing.Optional[int] = None
) -> None

nemo_gym.exporters.wandb.WandbExporter._log_rollouts(
    rollouts: list[dict[str, typing.Any]]
) -> None

nemo_gym.exporters.wandb.WandbExporter.setup() -> None

nemo_gym.exporters.wandb.WandbExporter.teardown() -> None