> 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.

# batch_size

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

## Description

A list of potential batch sizes during data loading.

* `batch_size` must be `<=2048`.

### Supported Task Types

* All

### Default Values

| run\_mode | Default Values |
| --------- | -------------- |
| FAST      | `[512, 1024]`  |
| NORMAL    | `[512, 1024]`  |
| BEST      | `[512, 1024]`  |