NVIDIA-NeMo / NVIDIA-NeMo/RL

[PyT DTensor]Qwen/Qwen3-30B-A3B fails due to assertion error

Open
#758 1 comment 0 reactions 0 assignees View on GitHub
bug t-pytdensor
Dominant language
Python
Stars
2k
Forks
561
Avg merge
4d 5h
Merged PRs (30d)
145

Description

**Describe the bug**

```
assert tokens_left_for_obs >= 0, (
```

**Steps/Code to reproduce bug**

```
uv run ./examples/run_grpo_math.py --config examples/configs/recipes/llm/grpo-qwen2.5-32b-32n8g-fsdp2tp8sp-actckpt.v3.yaml grpo.max_num_steps=1000000 grpo.val_period=10 policy.max_total_sequence_length=4096 policy.model_name=Qwen/Qwen3-30B-A3B policy.tokenizer.name=Qwen/Qwen3-30B-A3B policy.dtensor_cfg.tensor_parallel_size=4 policy.dtensor_cfg.sequence_parallel=False policy.dynamic_batching.enabled=True checkpointing.enabled=false logger.wandb_enabled=true logger.monitor_gpus=true
```

**Expected behavior**

A clear and concise description of what you expected to happen.

**Environment overview (please complete the following information)**

- Environment location: [Bare-metal, Docker, Cloud(specify cloud provider - AWS, Azure, GCP, Collab)]
- Method of install: [pip install or from source]. Please specify exact commands you used to install.
- If method of install is [Docker], provide `docker pull` & `docker run` commands used

**Environment details**

If NVIDIA docker image is used you don't need to specify these.
Otherwise, please provide:
- OS version
- PyTorch version
- Python version

**Additional context**

Add any other context about the problem here.
Example: GPU model

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.