Skip to content

fix: Enforce minimum packed sequence bin count and multiple of bin count#748

Merged
SahilJain314 merged 2 commits intomainfrom
sahilj/packing_multiple_fix
Jul 25, 2025
Merged

fix: Enforce minimum packed sequence bin count and multiple of bin count#748
SahilJain314 merged 2 commits intomainfrom
sahilj/packing_multiple_fix

Conversation

@SahilJain314
Copy link
Copy Markdown
Contributor

What does this PR do ?

Fixes bin counts being too low with GBS is low as reported by @ZhiyuLi-Nvidia and also #718.

Signed-off-by: Sahil Jain <sahilj@nvidia.com>
Signed-off-by: Sahil Jain <sahilj@nvidia.com>
@ZhiyuLi-Nvidia
Copy link
Copy Markdown
Contributor

Thank you @SahilJain314 test and verified your fix on my end as well.

Comment thread nemo_rl/data/packing/algorithms.py
Comment thread nemo_rl/distributed/batched_data_dict.py
@SahilJain314 SahilJain314 added this pull request to the merge queue Jul 25, 2025
@github-merge-queue github-merge-queue Bot removed this pull request from the merge queue due to failed status checks Jul 25, 2025
@SahilJain314 SahilJain314 added this pull request to the merge queue Jul 25, 2025
Merged via the queue into main with commit 9bdb75d Jul 25, 2025
25 of 27 checks passed
@SahilJain314 SahilJain314 deleted the sahilj/packing_multiple_fix branch July 25, 2025 22:55
chtruong814 pushed a commit that referenced this pull request Jul 25, 2025
…unt (#748)

Signed-off-by: Sahil Jain <sahilj@nvidia.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>
xxman-google pushed a commit to xxman-google/NeMo-RL that referenced this pull request Jul 28, 2025
rohitrango pushed a commit that referenced this pull request Jul 29, 2025
…unt (#748)

Signed-off-by: Sahil Jain <sahilj@nvidia.com>
wedu-nvidia pushed a commit to wedu-nvidia/RL that referenced this pull request Jul 30, 2025
xxman-google pushed a commit to xxman-google/NeMo-RL that referenced this pull request Jul 30, 2025
FannYYW pushed a commit to xxman-google/NeMo-RL that referenced this pull request Aug 5, 2025
FannYYW pushed a commit to xxman-google/NeMo-RL that referenced this pull request Aug 5, 2025
soodoshll pushed a commit to soodoshll/RL that referenced this pull request Aug 13, 2025
…unt (NVIDIA-NeMo#748)

Signed-off-by: Sahil Jain <sahilj@nvidia.com>
Signed-off-by: Qidong Su <qidongs@nvidia.com>
jiemingz pushed a commit that referenced this pull request Aug 20, 2025
…unt (#748)

Signed-off-by: Sahil Jain <sahilj@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants