Downloads · 30 days
62
8% of all-time downloads
RL4TG-Qwen2.5-3B-GRPO
tomhu/RL4TG-Qwen2.5-3B-GRPO
RL4TG-Qwen2.5-3B-GRPO is a text generation model from tomhu. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as apache-2.0.
This is an RL4TG checkpoint for Java unit-test generation.
All-time downloads
794
Public
Parameters
3.1B
55.6 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors6.2 GB · 100%
At a glance
- Task
- Text Generation
- Library
- transformers
- License
- apache-2.0
- Model type
- qwen2
- Access
- Public
- Created
- Jul 16, 2026
- Updated
- Jul 16, 2026
- SHA
- ce33dba7
Try a prompt
Base models
Model card
From the Hugging Face model README
Qwen2.5-3B + GRPO (2 epochs)
This is an RL4TG checkpoint for Java unit-test generation.
| Field | Value |
|---|---|
| Base model | Qwen/Qwen2.5-3B-Instruct |
| Method | Group Relative Policy Optimization (GRPO) |
| Teacher | None |
| Released checkpoint | step 98 (2 epochs) |
| Training data | 1,582 mutation-eligible Defects4J training samples |
| Prompt format | Qwen2.5 chat template |
| Global training seed | 42 |
Usage
from transformers import AutoModelForCausalLM, AutoTokenizer
model_id = "tomhu/RL4TG-Qwen2.5-3B-GRPO-2-Epochs"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(
model_id,
torch_dtype="auto",
device_map="auto",
)
The model is intended for research on Java unit-test generation. Generated tests must still be compiled and executed in the target project's own build environment.
<!-- RL4TG_CHECKPOINT_REVISIONS -->Checkpoint revisions
The 10-step trajectory checkpoints are available as Git revisions in this repo.
The main revision is the final step-98 model.
| Revision | Training step | Load argument |
|---|---|---|
step-10 | 10 | revision="step-10" |
step-20 | 20 | revision="step-20" |
step-30 | 30 | revision="step-30" |
step-40 | 40 | revision="step-40" |
step-50 | 50 | revision="step-50" |
step-60 | 60 | revision="step-60" |
step-70 | 70 | revision="step-70" |
step-80 | 80 | revision="step-80" |
step-90 | 90 | revision="step-90" |
- Task
- Text Generation
- Library
- transformers
- Type
- qwen2
- License
- apache-2.0
- Created
- Jul 16, 2026
- Updated
- Jul 16, 2026