/venv/main/lib/python3.12/site-packages/trl/__init__.py:203: UserWarning: TRL currently supports vLLM versions: 0.10.2, 0.11.0, 0.11.1, 0.11.2. You have version 0.13.0 installed. We recommend installing a supported version to avoid compatibility issues. if is_vllm_available(): Loading config from: configs/runtime_config.yaml Random seed set to: 42 ============================================================ Training Configuration: Framework: accelerate Method: sft ============================================================ W&B Project: aimo3-exp-1-train W&B Run: qwen3-0.6b-sft-lora-20260113_063547-003 wandb: Currently logged in as: vinoku (vinokuteam) to https://api.wandb.ai. Use `wandb login --relogin` to force relogin wandb: WARNING Using a boolean value for 'reinit' is deprecated. Use 'return_previous' or 'finish_previous' instead. wandb: setting up run 0pkyx3cq wandb: Tracking run with wandb version 0.23.1 wandb: Run data is saved locally in /workspace/distributed-sft-rl-inference-pipeline/wandb/run-20260112_223606-0pkyx3cq wandb: Run `wandb offline` to turn off syncing. wandb: Syncing run qwen3-0.6b-sft-lora-20260113_063547-003 wandb: ⭐️ View project at https://wandb.ai/vinokuteam/aimo3-exp-1-train wandb: 🚀 View run at https://wandb.ai/vinokuteam/aimo3-exp-1-train/runs/0pkyx3cq wandb: Detected [openai] in use. wandb: Use W&B Weave for improved LLM call tracing. Install Weave with `pip install weave` then add `import weave` to the top of your script. wandb: For more information, check out the docs at: https://weave-docs.wandb.ai/ W&B Run: qwen3-0.6b-sft-lora-20260113_063547-003 W&B URL: https://wandb.ai/vinokuteam/aimo3-exp-1-train/runs/0pkyx3cq W&B logging enabled (ID: 0pkyx3cq) Loading model: Qwen/Qwen3-0.6B ZeRO-0 mode (no HfDeepSpeedConfig needed) `torch_dtype` is deprecated! Use `dtype` instead! Building datasets... Loading dataset from HuggingFace: HAD653/GSM8K-OpenMath-MathReason-13k Train size: 13857, Test size: 416 Dataset size after dropping NaN: 13857 Dataset size after dropping NaN: 416 Final train size: 13857, test size: 416 ============================================================ SAMPLE TRAINING DATA (first example): ============================================================ <|im_start|>system You are a helpful and reflective maths assistant. Please reason step by step to put the answer in \boxed{}.<|im_end|> <|im_start|>user Albert is wondering how much pizza he can eat in one day. He buys 2 large pizzas and 2 small pizzas. A large pizza has 16 slices and a small pizza has 8 slices. If he eats it all, how many pieces does he eat that day?<|im_end|> <|im_start|>assistant Problem: Albert buys 2 large pizzas (16 slices each) and 2 small pizzas (8 slices each). Find total slices eaten. Reasoning: 1. Large pizza slices = 2*16 = 32. 2. Small pizza slices = 2*8 = 16. 3. Total slices = 32+16 = 48. Answer: 48 \boxed{48}<|im_end|> ============================================================ Creating LoRA configuration... Using Accelerate + SFT training pipeline Initializing SFTTrainer... Adding EOS to train dataset: 0%| | 0/13857 [00:00= best: 2.0518) Upload complete: https://huggingface.co/vinoku89/aimo3-exp-1 Uploaded logs/runtime_config_20260113_063547.yaml to vinoku89/aimo3-exp-1 (main branch) ============================================================ Model pushed successfully! ============================================================ URL: https://huggingface.co/vinoku89/aimo3-exp-1 Run Tag: qwen0.6b-sft-e1-13.9k-loss2.05-20260112-003 Tags: qwen0.6b-sft-e1-13.9k-loss2.05-20260112-003, latest, best (if lowest loss) To load this specific run: from peft import PeftModel model = PeftModel.from_pretrained('base_model', 'vinoku89/aimo3-exp-1', revision='qwen0.6b-sft-e1-13.9k-loss2.05-20260112-003') To load the best run: model = PeftModel.from_pretrained('base_model', 'vinoku89/aimo3-exp-1', revision='best') To load the latest run: model = PeftModel.from_pretrained('base_model', 'vinoku89/aimo3-exp-1', revision='latest') ============================================================ wandb: updating run metadata wandb: uploading output.log; uploading wandb-summary.json wandb: uploading output.log wandb: uploading config.yaml wandb: uploading summary, console lines 54-87 wandb: wandb: Run history: wandb: eval/entropy ▁ wandb: eval/loss ▁ wandb: eval/mean_token_accuracy ▁ wandb: eval/num_tokens ▁ wandb: eval/runtime ▁ wandb: eval/samples_per_second ▁ wandb: eval/steps_per_second ▁ wandb: train/entropy ▁ wandb: train/epoch ▁▁▁ wandb: train/global_step ▁▁▁ wandb: +5 ... wandb: wandb: Run summary: wandb: eval/entropy 0.92732 wandb: eval/loss 2.05312 wandb: eval/mean_token_accuracy 0.66951 wandb: eval/num_tokens 21152 wandb: eval/runtime 13.5735 wandb: eval/samples_per_second 30.648 wandb: eval/steps_per_second 15.324 wandb: total_flos 67838115053568.0 wandb: train/entropy 0.94746 wandb: train/epoch 0.00577 wandb: +10 ... wandb: wandb: 🚀 View run qwen3-0.6b-sft-lora-20260113_063547-003 at: https://wandb.ai/vinokuteam/aimo3-exp-1-train/runs/0pkyx3cq wandb: ⭐️ View project at: https://wandb.ai/vinokuteam/aimo3-exp-1-train wandb: Synced 5 W&B file(s), 0 media file(s), 0 artifact file(s) and 0 other file(s) wandb: Find logs at: ./wandb/run-20260112_223606-0pkyx3cq/logs Training completed successfully! [W112 22:37:40.785124394 AllocatorConfig.cpp:28] Warning: PYTORCH_CUDA_ALLOC_CONF is deprecated, use PYTORCH_ALLOC_CONF instead (function operator()) [W112 22:37:42.435659668 AllocatorConfig.cpp:28] Warning: PYTORCH_CUDA_ALLOC_CONF is deprecated, use PYTORCH_ALLOC_CONF instead (function operator())