SPACE Qwen3-4B (Ultrachat)
Self-play SPACE (Wang et al., NeurIPS 2025) run on Qwen3-4B SFT
(HoangTran223/qwen3_4b_sft_ultrachat200k_20260818_170154).
- Prompts: UltraChat 50k subset of
HuggingFaceH4/ultrachat_200k - Loss: SPACE (
mu=1.0), RMSProp, batch 2 × grad accum 2,max_length=1024 - LR:
5e-7(ite0–1),1e-7(ite2–3)
All 4 iterations finished. ite3/ is the final policy.
from transformers import AutoModelForCausalLM
m = AutoModelForCausalLM.from_pretrained("HoangTran223/SPACE_Qwen3-4B", subfolder="ite3")
Layout
ite0/,ite1/,ite2/,ite3/: finished checkpoints + generatedtrain.jsonldata/Ultrachat50k/train.jsonl: prompts used for generationscripts/SPACE_full_qwen3_4b.sh: launch scriptlogs/: wandb offline runs, Hydra configs, tmux stdout