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

# num_heads

#### `num_heads: <list[int]>` (Optional)

#### Description

Specifies the number of attention heads in the GraphTransformer model.

#### Supported Task Types

* All except Link Prediction

#### Default Values

| [run\_mode](/run-mode) | Default Value |
| :--------------------- | :------------ |
| FAST                   | `[8]`         |
| NORMAL                 | `[8]`         |
| BEST                   | `[8]`         |