Skip to content

feat: add capability to set min/max eps separately as proposed in the…#95

Merged
parthchadha merged 3 commits intomainfrom
pchadha/dapo-eps
Mar 31, 2025
Merged

feat: add capability to set min/max eps separately as proposed in the…#95
parthchadha merged 3 commits intomainfrom
pchadha/dapo-eps

Conversation

@parthchadha
Copy link
Copy Markdown
Contributor

… DAPO paper

What does this PR do ?

Allows users to set min/max clipping threshold as proposed in the DAPO paper.

Screenshot 2025-03-29 at 8 08 00 AM

Issues

List issues that this PR closes (syntax):

Usage

  • You can potentially add a usage example below
# Add a code snippet demonstrating how to use this 

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

  • ...

… DAPO paper

Signed-off-by: Parth Chadha <pchadha@nvidia.com>
Copy link
Copy Markdown
Contributor

@SahilJain314 SahilJain314 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Modified a bit of documentation on the loss, but otherwise LGTM

Signed-off-by: Parth Chadha <pchadha@nvidia.com>
@parthchadha parthchadha enabled auto-merge (squash) March 31, 2025 18:59
@parthchadha parthchadha merged commit a2c583e into main Mar 31, 2025
10 of 11 checks passed
@parthchadha parthchadha deleted the pchadha/dapo-eps branch March 31, 2025 19:11
yfw pushed a commit that referenced this pull request Apr 2, 2025
#95)

Signed-off-by: Parth Chadha <pchadha@nvidia.com>
Co-authored-by: Sahil Jain <48468750+SahilJain314@users.noreply.github.com>
Signed-off-by: Yi-Fu Wu <yifu.wu@gmail.com>
KiddoZhu pushed a commit that referenced this pull request May 6, 2025
#95)

Signed-off-by: Parth Chadha <pchadha@nvidia.com>
Co-authored-by: Sahil Jain <48468750+SahilJain314@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants