feat: support grpo_remote_rm and related cookbook #189
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Description
Implemented the GRPO with remote reward model and added cookbook for both Chinese and English.
Motivation and Context
This update consists of two main components:
Implemented
grpo_remote_rm.py
based on the existingppo_remote_rm.py
, providing support for grpo with remote reward.Created a comprehensive
text_to_text_grpo.ipynb
cookbook that demonstrates:Test
Please test your changes by running the following command:
cd scripts bash test/test_text_to_text.sh ./opt PATH_TO_OUTPUT_ROOT_DIR
Here,
./opt
is the directory containing the test scripts for theopt
model, andPATH_TO_OUTPUT_ROOT_DIR
is the path to the output root directory. The test scripts will save ~1GB data to the output root directory and delete it after the test. Please ensure you have enough space on your disk.Lint
Please run the following command in the root directory to check your code style:
Types of changes
What types of changes does your code introduce? Put an
x
in all the boxes that apply:Checklist
Go over all the following points, and put an
x
in all the boxes that apply.If you are unsure about any of these, don't hesitate to ask. We are here to help!