Search-R1-Qwen3-8B-Process-GRPO-PerTurnNorm

A Qwen/Qwen3-8B policy trained with GRPO on Search-R1 multi-turn retrieval QA with Process-GRPO: an Olmo-3-7B-Think verifier scores each turn, normalized per (group, turn-position).

The repository root holds the final policy from an earlier upload; every checkpoint added since lives in a step_<N>/ subfolder.

The root model and the step_{50..350} subfolders come from an earlier upload whose local copies have since been deleted; the remaining subfolders were added from local exports. Together they cover every 10 steps from 10 to 400.

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained("wckwan/Search-R1-Qwen3-8B-Process-GRPO-PerTurnNorm")
tokenizer = AutoTokenizer.from_pretrained("wckwan/Search-R1-Qwen3-8B-Process-GRPO-PerTurnNorm")

# An intermediate checkpoint
model_step = AutoModelForCausalLM.from_pretrained("wckwan/Search-R1-Qwen3-8B-Process-GRPO-PerTurnNorm", subfolder="step_10")

Checkpoints

  • step_10/
  • step_20/
  • step_30/
  • step_40/
  • step_60/
  • step_70/
  • step_80/
  • step_90/
  • step_110/
  • step_120/
  • step_130/
  • step_140/
  • step_160/
  • step_170/
  • step_180/
  • step_190/
  • step_210/
  • step_220/
  • step_230/
  • step_240/
  • step_260/
  • step_270/
  • step_280/
  • step_290/
  • step_310/
  • step_320/
  • step_330/
  • step_340/
  • step_360/
  • step_370/
  • step_380/
  • step_390/
  • step_50/ (uploaded earlier; no local copy remains)
  • step_100/ (uploaded earlier; no local copy remains)
  • step_150/ (uploaded earlier; no local copy remains)
  • step_200/ (uploaded earlier; no local copy remains)
  • step_250/ (uploaded earlier; no local copy remains)
  • step_300/ (uploaded earlier; no local copy remains)
  • step_350/ (uploaded earlier; no local copy remains)

Raw FSDP checkpoints

fsdp/global_step_<N>/ holds the unmerged verl FSDP checkpoint (sharded fp32 model state, optimizer state and extra state) for step(s) 400. These are for resuming training, not for inference — use the merged exports above to load a policy.

Downloads last month
430
Safetensors
Model size
8B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for wckwan/Search-R1-Qwen3-8B-Process-GRPO-PerTurnNorm

Finetuned
Qwen/Qwen3-8B
Finetuned
(1982)
this model