> For clean Markdown of any page, append .md to the page URL.
> For a complete documentation index, see https://docs.nvidia.com/nemo/automodel/llms.txt.
> For AI client integration (Claude Code, Cursor, etc.), connect to the MCP server at https://docs.nvidia.com/nemo/automodel/_mcp/server.

# nemo_automodel.components.loggers.wandb_utils

## Module Contents

### Functions

| Name                                                                                                        | Description                                       |
| ----------------------------------------------------------------------------------------------------------- | ------------------------------------------------- |
| [`init_wandb_run`](#nemo_automodel-components-loggers-wandb_utils-init_wandb_run)                           | Initialize a W\&B run from a config dict.         |
| [`suppress_wandb_log_messages`](#nemo_automodel-components-loggers-wandb_utils-suppress_wandb_log_messages) | Patches wandb logger to suppress upload messages. |

### API

```python
nemo_automodel.components.loggers.wandb_utils.init_wandb_run(
    wandb_cfg: dict,
    full_config: dict,
    default_name: str = ''
)
```

Initialize a W\&B run from a config dict.

A recipe-agnostic counterpart to recipe-local `build_wandb` helpers: it
takes the already-resolved `wandb` kwargs (project, entity, name, ...) and
the full run config to record, and falls back to `default_name` when no
run name is given.

**Parameters:**

Keyword args forwarded to `wandb.init` (e.g. project, entity, name).

The full resolved config to store as the run's `config`.

Run name to use when `wandb_cfg` does not set one.

**Returns:**

The initialized `wandb.Run`.

```python
nemo_automodel.components.loggers.wandb_utils.suppress_wandb_log_messages() -> None
```

Patches wandb logger to suppress upload messages.

These occur usually on KeyboardInterrupt or program crash.

To print the log url:
run = wandb.init(...)
print(run.url)