Views
No views yet
Qwen3.5-4B-GRPO-01R-2048-ifeval-v0-think.| Algorithm | GRPO (adv_estimator=grpo), 4 rollouts per prompt |
| Steps | 200 |
| Train batch size | 128 prompts |
| Mini / micro batch | 64 / 2 per GPU |
| Max prompt / response length | 1024 / 2048 tokens |
| Learning rate | 1e-6 |
| KL | use_kl_loss=True, kl_loss_coef=0.001, low_var_kl; no KL in reward |
| Entropy coefficient | 0 |
| Rollout | vLLM, TP=1, temperature per verl defaults |
config.json declares dtype: bfloat16, so transformers casts to
bf16 on load by default. Pass dtype="float32" if you want the stored precision.1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = "ZetaRRR/Qwen3.5-4B-VerIH-step200"
4tok = AutoTokenizer.from_pretrained(model_id)
5model = AutoModelForCausalLM.from_pretrained(model_id, dtype="auto", device_map="auto")
6
7messages = [{"role": "user", "content": "Write a haiku about gradients. Use no commas."}]
8inputs = tok.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to(model.device)
9out = model.generate(inputs, max_new_tokens=2048)
10print(tok.decode(out[0][inputs.shape[-1]:], skip_special_tokens=True))