Qwen3-8B ToolUse TR Methods
Collection
Qwen3-8B teacher-regularized RL method collection for the ToolUse dataset. Includes GRPO-TR, RLSD-TR, SDPO-TR, and SRPO-TR models. • 5 items • Updated
How to use SeongryongJung/Qwen3-8B-ToolUse-SDPO-TR with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("SeongryongJung/Qwen3-8B-ToolUse-SDPO-TR")
model = AutoModelForCausalLM.from_pretrained("SeongryongJung/Qwen3-8B-ToolUse-SDPO-TR", device_map="auto")This repository contains the Qwen3-8B Tool use SDPO-TR run trained with batch size 32.
The repository root contains the best validation checkpoint (global_step_40), selected by validation mean@16.
checkpoints/last/ contains the final checkpoint.
| Dataset | Method | Base model | Train batch size | Best val mean@16 | Best checkpoint | Final val mean@16 | Final checkpoint |
|---|---|---|---|---|---|---|---|
| Tool use | SDPO-TR | Qwen3-8B | 32 | 66.36% | 40 | 61.40% | 100 |
Raw result files:
results/validation_mean16.csvresults/training_scores.csvartifacts/config.yamlartifacts/wandb-summary.json| Hyperparameter | Value |
|---|---|
| Base model | Qwen/Qwen3-8B |
| Method | SDPO-TR |
| Train batch size | 32 |
| Train max samples | 3200 |
| Total training steps | 100 |
| Save frequency | 10 |
| Test frequency | 10 |
| Teacher regularization | trust-region |
| Teacher update rate | 0.1 |
| Rollout samples per prompt | 8 |
| Validation samples per prompt | 16 |
| Learning rate | 1e-5 |
| vLLM GPU memory utilization | 0.8 |
from transformers import AutoModelForCausalLM, AutoTokenizer
repo_id = "SeongryongJung/Qwen3-8B-ToolUse-SDPO-TR"
tokenizer = AutoTokenizer.from_pretrained(repo_id, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(
repo_id,
torch_dtype="auto",
device_map="auto",
trust_remote_code=True,
)