THUDM / THUDM/slime

[Bug] Resume training results in very high rollout logprobs diff and very low pass@1/raw_reward for a few steps

Open
#1,695 0 comments 1 reaction 0 assignees View on GitHub

Nobody has claimed this yet.

bug
Dominant language
Python
Stars
8.5k
Forks
1.3k
Avg merge
5h 36m
Merged PRs (30d)
22

Description

Bug Description
Image
Steps to Reproduce
  1. train qwen-30b-a3b-thinking-2507
  2. load checkpoint:
CKPT_ARGS=(
   --hf-checkpoint /tmp/instance_storage/${MODEL_CONFIG}/
   --ref-load /tmp/instance_storage/${MODEL_CONFIG}_torch_dist/
   --load /tmp/instance_storage/checkpoints/${MODEL_CONFIG}_slime/
   --save /tmp/instance_storage/checkpoints/${MODEL_CONFIG}_slime/
   --save-interval 20
)
Expected Behavior

no numerical error

Actual Behavior

very high log probs diff / very low pass@1

Environment
  • slime version: c25982336374bcdcb44635410771bd78166684c1
  • Python version: 3.12
  • PyTorch version: 2.9.1
  • CUDA/ROCm version:
  • GPU type and count: 24xH200
  • OS:
  • SGLang version (if relevant):
  • Megatron-LM version (if relevant):
Logs

Additional Context

No response

Pre-submission Checklist
  • I have read the CONTRIBUTING.md and understand the collaboration scope.
  • I have read the documentation and my issue is not addressed there.
  • I have searched for existing issues and this is not a duplicate.
  • I have provided a minimal, reproducible example.

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by reproducing the resume flow with the provided checkpoint arguments and compare rollout logprobs, pass@1, and raw_reward before and after loading the checkpoint. The issue is resolved when resumed training no longer produces the reported numerical discrepancy or temporary metric collapse.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.