keeping save steps at 2k

scripts/train.py

@@ -78,7 +78,7 @@ def main():
         num_train_epochs=3,
         logging_dir="./logs",
         logging_steps=10,
-        save_steps=5,
+        save_steps=2000,
         save_total_limit=2,
         bf16=True,
         gradient_checkpointing=False,  # Disabled for now