Downloads · 30 days
0
logan7000/openr1-gemma3-4b-gt-mllm-repro
openr1-gemma3-4b-gt-mllm-repro is a reinforcement learning model from logan7000. Use it for the reinforcement learning task on the model card, and read the license before you ship it in a product. The card lists the license as gemma.
Checkpoints from a full 1-epoch run, saved every 20 steps, so the whole training trajectory can be re-evaluated rather than only its endpoint. Trained with mllm-repro on lmms-lab/multimodal-open-r1-8k-verified.
Downloads · 30 days
0
Access
Public
Updated Aug 8, 2026
Repo size
421 GB
Likes
0
Public
Click a slice to open those files.
.safetensors430 GB · 100%
From the Hugging Face model README
Checkpoints from a full 1-epoch run, saved every 20 steps, so the whole training
trajectory can be re-evaluated rather than only its endpoint. Trained with
mllm-repro on
lmms-lab/multimodal-open-r1-8k-verified.
Weights only -- ZeRO-3 optimizer shards are stripped, so these are for evaluation and further fine-tuning, not for resuming this run.
The in-loop curve is flat. It starts at 0.329 and ends at 0.362, and every one of the 48 evaluations falls in 0.296-0.434 with no trend. Compare the same recipe on Qwen2.5-VL-3B, which climbs 0.283 -> 0.55 within 250 steps: that one has a shape, this one does not. GT training on this dataset produced no measurable gain for this model.
The maximum at step 760 is 0.06 above its neighbours (740 scores 0.375, 780 scores 0.382) while a single measurement on 150 problems carries a standard error of about 0.041, so that peak is noise rather than a checkpoint worth selecting.
| base | google/gemma-3-4b-it |
| attention | sdpa (FA2 crashes Gemma-3 in this configuration) |
| effective batch | 8 prompts / 64 completions per step, num_generations=8 |
| layout | 4x80GB, bs=2 x ga=8 x 4proc. The 8-GPU recipe is bs=1 x ga=8 x 8proc; both give 8 prompts per step, which is the quantity to hold fixed |
| lr | 1e-6, cosine with min_lr_rate 0.1, warmup 0.03 |
| generation | max_completion_length=1024, temperature 1.0 |
| steps | 961 = 1 epoch |
| in-loop eval | MathVista-150 every 20 steps |
Maximum 0.4342 at step 760. The eval set is 150 problems, so one problem is 0.67 points and a single measurement carries a standard error of roughly 0.041; score candidates on a real benchmark suite before picking one.
| step | eval_reward |
|---|---|
| 20 | 0.3289 |
| 40 | 0.3947 |
| 60 | 0.3289 |
| 80 | 0.3553 |
| 100 | 0.2961 |
| 120 | 0.3355 |
| 140 | 0.3487 |
| 160 | 0.3224 |
| 180 | 0.3947 |
| 200 | 0.4079 |
| 220 | 0.3882 |
| 240 | 0.3421 |
| 260 | 0.3882 |
| 280 | 0.3289 |
| 300 | 0.3618 |
| 320 | 0.3816 |
| 340 | 0.3618 |
| 360 | 0.3618 |
| 380 | 0.3618 |
| 400 | 0.3618 |
| 420 | 0.3947 |
| 440 | 0.3947 |
| 460 | 0.3750 |
| 480 | 0.3421 |
| 500 | 0.3487 |
| 520 | 0.3355 |
| 540 | 0.3224 |
| 560 | 0.3487 |
| 580 | 0.3750 |
| 600 | 0.3553 |
| 620 | 0.3684 |
| 640 | 0.3355 |
| 660 | 0.3618 |
| 680 | 0.3421 |
| 700 | 0.3421 |
| 720 | 0.3289 |
| 740 | 0.3750 |
| 760 | 0.4342 |
| 780 | 0.3816 |
| 800 | 0.3882 |
| 820 | 0.3816 |
| 840 | 0.3684 |
| 860 | 0.3684 |
| 880 | 0.3487 |
| 900 | 0.4079 |
| 920 | 0.4079 |
| 940 | 0.4013 |
| 960 | 0.3618 |