Views
No views yet
experiment_name=grpo_omr_4node_full_v1_8b_base_perf, global_step_25 (the keeper).0.668 @ step 25, then collapses to 0.499 by step 125 — cold-start RL on 8B-OMR is unstable without exploration. The keeper is global_step_25, the only checkpoint before degradation began. (The omr-8b-grpo-ppexplore sibling fixes this collapse and reaches 0.714.)overall-6 = unweighted mean. Metric = accuracy.global_step_25 (peak):| metric | overall-6 | mmmu | mathvista | mathverse | wemath | charxiv | dynamath |
|---|---|---|---|---|---|---|---|
| baseline peak @25 | 0.6681 | 0.6311 | 0.8019 | 0.8404 | 0.7437 | 0.3930 | 0.5986 |
| stock Qwen3-VL-8B ckpt-0 | 0.659 | 0.629 | 0.811 | 0.824 | 0.723 | 0.396 | 0.570 |
| baseline @125 (collapse) | 0.4994 | 0.5021 | 0.6796 | 0.6995 | 0.6184 | 0.2010 | 0.2958 |
Qwen/Qwen3-VL-8B-Instruct (cold-start, resume_mode=auto).ngquangtrung57/verl@videorl-mods. Fully-async GRPO: FSDP2 trainer + vLLM rollouter.score = 0.8·accuracy + 0.2·format (FORMAT_WEIGHT=0.2, FORMAT_MIN_THINK_CHARS=100). No KL penalty.omr-8b-grpo-ppexplore).ppo_mini_batch_size=16 × require_batches=4 × rollout.n=8 = 512 trajectories/step.1e-6, warmup 25 steps; total_epochs=2; clip_ratio 0.2 / 0.3 (clip_c=10.0); max_prompt_length=2048, max_response_length=16384; enforce_eager=true; gpu_memory_utilization=0.75; staleness 0.5.verl_fully_async (entity quangtrung5705-nanyang-technological-university-singapore):
https://wandb.ai/quangtrung5705-nanyang-technological-university-singapore/verl_fully_async/runs/gkiiopepomr-8b-grpo-ppexplore (the winner). It only marginally beats the zero-shot base (0.668 vs 0.659) at its peak and is unstable (collapses to 0.499 if trained past ~step 100); for actual use prefer the ppexplore checkpoint. Math / visual-reasoning image QA, <think>…</think> then-answer format. No safety/RLHF alignment beyond the base.1from transformers import AutoModelForImageTextToText, AutoProcessor
2from PIL import Image
3
4model_id = "ngqtrung/omr-8b-grpo-base"
5model = AutoModelForImageTextToText.from_pretrained(model_id, dtype="auto", device_map="auto")
6processor = AutoProcessor.from_pretrained(model_id)
7
8messages = [{
9 "role": "user",
10 "content": [
11 {"type": "image", "image": Image.open("problem.png")},
12 {"type": "text", "text": "Solve the problem. Think step by step inside <think>...</think>, then give the final answer."},
13 ],
14}]
15inputs = processor.apply_chat_template(
16 messages, add_generation_prompt=True, tokenize=True,
17 return_dict=True, return_tensors="pt"
18).to(model.device)
19out = model.generate(**inputs, max_new_tokens=2048)
20print(processor.batch_decode(out[:, inputs["input_ids"].shape[1]:], skip_special_tokens=True)[0])ngquangtrung57/verl@videorl-mods; fully-async GRPO (FSDP2 + vLLM).docs/experiments_summary_8b.md).