Skip to content

feat: Phase 4.B on-policy vision RL - #57

Merged
ajwharton merged 1 commit into
mainfrom
feat/phase4b-vision-rl
Jul 30, 2026
Merged

ajwharton merged 1 commit into
mainfrom
feat/phase4b-vision-rl

Conversation

@ajwharton

Copy link
Copy Markdown
Owner

Summary

Phase 4.B product path for on-policy vision/robot RL under four verbs:

  1. Multimodal sample — run_grpo prompts may be ModelInput (text tokens + ImageRefChunk)
  2. Vision rewards — keyword / rubric / action-bin on detokenized text (vision_rewards)
  3. Stage queue — run_vision_rl_queue (caption → hazard ladder toy)

Also fixes reward_fn always receiving "" (now gets detokenized completion when provided).

Test plan

  • 268 tests pass; ruff clean
  • python scripts/vision_rl_demo.py --steps 2
  • Operator: forge vision GRPO on real VLM + media store

- run_grpo accepts ModelInput prompts and per-prompt rewards; passes
  detokenized text into reward_fn; job=vision_grpo metrics + n_image_refs
- vision_rewards: keyword, phrase, action-bin, weighted rubric scorers
- run_vision_grpo + run_vision_rl_queue curriculum on shared LoRA
- catalog vision_rl_on_policy; smoke scripts/vision_rl_demo.py
@ajwharton
ajwharton merged commit 4507266 into main Jul 30, 2026
1 check passed
@ajwharton
ajwharton deleted the feat/phase4b-vision-rl branch July 30, 2026 14:39
ajwharton added a commit that referenced this pull request Jul 30, 2026
Keep Phase 4.B as done on main; retain FPS smoke + dogfood next steps.
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.

1 participant