API ReferenceFull Library ReferenceNemo GymNemo GymExporters
nemo_gym.exporters.wandb
nemo_gym.exporters.wandb
This page is for version 0.5.1.
For other versions, use one of these documentation indexes:
- Main (default): https://docs.nvidia.com/nemo/gym/main/llms.txt
- 0.7.0: https://docs.nvidia.com/nemo/gym/v0.7.0/llms.txt
- 0.6.0: https://docs.nvidia.com/nemo/gym/v0.6.0/llms.txt
- 0.5.1: https://docs.nvidia.com/nemo/gym/v0.5.1/llms.txt
- 0.5.0: https://docs.nvidia.com/nemo/gym/v0.5.0/llms.txt
- 0.4.0: https://docs.nvidia.com/nemo/gym/v0.4.0/llms.txt
- 0.3.0: https://docs.nvidia.com/nemo/gym/v0.3.0/llms.txt
- 0.2.1: https://docs.nvidia.com/nemo/gym/v0.2.1/llms.txt
Module Contents
Classes
| Name | Description |
|---|---|
WandbExporter | Weights & Biases backend. |
API
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