Skip to content

fix: Fix the default setting in Nemo Gym Nano v3 recipe config#2857

Open
snowmanwwg wants to merge 4 commits into
mainfrom
fix/nemo-gym-nanov3-draft-default
Open

fix: Fix the default setting in Nemo Gym Nano v3 recipe config#2857
snowmanwwg wants to merge 4 commits into
mainfrom
fix/nemo-gym-nanov3-draft-default

Conversation

@snowmanwwg

Copy link
Copy Markdown
Contributor

What does this PR do ?

This PR disables the draft model for default setting in https://github.com/NVIDIA-NeMo/RL/blob/main/examples/nemo_gym/grpo_nanov3.yaml
. It mirrors what was done in https://github.com/NVIDIA-NeMo/RL/blob/main/examples/configs/grpo_math_1B.yaml#L258

Issues

Without this fix. Running this https://github.com/NVIDIA-NeMo/RL/blob/main/examples/nemo_gym/grpo_nanov3.yaml
gives an KeyError: 'draft' at config.policy["draft"]["enabled"] (https://github.com/NVIDIA-NeMo/RL/blob/main/examples/run_grpo.py#L104)

Before your PR is "Ready for review"

Pre checks:

  • Make sure you read and followed Contributor guidelines
  • Did you write any new necessary tests?
  • Did you run the unit tests and functional tests locally? Visit our Testing Guide for how to run tests
  • Did you add or update any necessary documentation? Visit our Document Development Guide for how to write, build and test the docs.

Additional Information

  • ...

Signed-off-by: Wenwen Gao <wenweng@cw-dfw-cs-001-vscode-01.cm.cluster>
@snowmanwwg snowmanwwg requested a review from a team as a code owner June 17, 2026 05:07
@copy-pr-bot

copy-pr-bot Bot commented Jun 17, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

Wenwen Gao added 2 commits June 16, 2026 22:58
Signed-off-by: Wenwen Gao <wenweng@cw-dfw-cs-001-vscode-01.cm.cluster>
Signed-off-by: Wenwen Gao <wenweng@cw-dfw-cs-001-vscode-01.cm.cluster>
@snowmanwwg snowmanwwg changed the title fix: disable draft config in Nemo Gym Nano v3 recipe default setting fix: Fix the default setting in Nemo Gym Nano v3 recipe config Jun 17, 2026
@terrykong terrykong enabled auto-merge (squash) June 17, 2026 07:15
@terrykong terrykong added the CI:Lfast Runs a fast test suite and re-use nightly `main` container (but sync dependencies to PRs version) label Jun 17, 2026
@terrykong

Copy link
Copy Markdown
Collaborator

/ok to test 324d965

@terrykong

Copy link
Copy Markdown
Collaborator

/ok to test 8a5c7bb

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CI:Lfast Runs a fast test suite and re-use nightly `main` container (but sync dependencies to PRs version)

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants