Skip to content

(feat): agentic rl stepwise learning GiGPO, support for multi-modal distill - #136

Merged
PanAndy merged 8 commits into
mainfrom
sync/sync_to_github_0810
Aug 10, 2025
Merged

(feat): agentic rl stepwise learning GiGPO, support for multi-modal distill#136
PanAndy merged 8 commits into
mainfrom
sync/sync_to_github_0810

Conversation

@PanAndy

@PanAndy PanAndy commented Aug 10, 2025

Copy link
Copy Markdown
Collaborator

(fix): fix agg loss bugs.
(docs): Add documentation for the distillation pipeline and update da…
(fix): fix seed in data generator.
(chore): update dpo_config.yaml file.
(feat): agentic add gigpo.
(docs): add documentation for supporting new models.
(feat): add support for multi-modal distill.

@CLAassistant

CLAassistant commented Aug 10, 2025

Copy link
Copy Markdown

CLA assistant check
All committers have signed the CLA.

@PanAndy

PanAndy commented Aug 10, 2025

Copy link
Copy Markdown
Collaborator Author

In this PR, we added StepEnvManager and implemented the GiGPO algorithm.

We removed the explicit addition of and tokens in the response. This was because we found that adding could introduce abnormal logprobs values, and this extra text should not be included in the loss calculation.
Upon verification, even after removing the explicit addition of , the model was still able to learn the desired format.

The training curves are shown below:
image

NaN appeared in the later stages of stepwise training.
We are very curious why tokens with such small probabilities are being sampled, and will continue to investigate this issue.

image

@PanAndy
PanAndy merged commit 30ec292 into main Aug 10, 2025
6 checks passed
@PanAndy
PanAndy deleted the sync/sync_to_github_0810 branch September 24, 2025 07:38
@ZhangXcho

Copy link
Copy Markdown

能否提供gigpo的训练入口呢

@HuangJoJo

Copy link
Copy Markdown
Collaborator

能否提供gigpo的训练入口呢

可从qwen2.5-0.5B-agentic_sokoban示例入手

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.

8 participants