Upload train_sft.py with huggingface_hub

train_sft.py

@@ -138,6 +138,8 @@ def main():
     ap.add_argument("--lr", type=float, default=2e-4)
     ap.add_argument("--max-steps", type=int, default=-1)
     ap.add_argument("--eval-frac", type=float, default=0.03)
+    ap.add_argument("--eval-steps", type=int, default=50)
+    ap.add_argument("--save-steps", type=int, default=100)
     ap.add_argument("--project", default="gemma4-gui-agent")
     ap.add_argument("--run-name", default="gemma4-e4b-lora")
     ap.add_argument("--private", action="store_true", default=True)
@@ -181,9 +183,9 @@ def main():
         lr_scheduler_type="cosine",
         logging_steps=5,
         eval_strategy="steps",
-        eval_steps=50,
+        eval_steps=args.eval_steps,
         save_strategy="steps",
-        save_steps=100,
+        save_steps=args.save_steps,
         save_total_limit=2,
         bf16=True,
         gradient_checkpointing=True,