Upload train_math.py

train_math.py

@@ -75,7 +75,7 @@ def prepare_datasets():
 
 
 def main():
-    trackio.init(project="qwen3-math-sft", run_name="qwen3-1.7b-math-lora")
+    run = trackio.init(project="qwen3-math-sft", name="qwen3-1.7b-math-lora")
     trackio.log_params({
         "model": MODEL_NAME, "epochs": EPOCHS, "lr": LR,
         "batch_size": PER_DEVICE_BATCH, "gradient_accumulation": GRADIENT_ACCUMULATION,
@@ -125,7 +125,7 @@ def main():
     trainer.save_model(OUTPUT_DIR)
     trainer.push_to_hub()
 
-    trackio.log_metrics({"training_status": "completed"})
+    trackio.log({"training_status": "completed"})
     trackio.finish()
     print("Training complete!")