/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_061956-001 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 3fkbe45r 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_222013-3fkbe45r wandb: Run `wandb offline` to turn off syncing. wandb: Syncing run qwen3-0.6b-sft-lora-20260113_061956-001 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/3fkbe45r 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_061956-001 W&B URL: https://wandb.ai/vinokuteam/aimo3-exp-1-train/runs/3fkbe45r W&B logging enabled (ID: 3fkbe45r) 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