fix: Update default training steps to 300 (GoogleCloudPlatform#10454)

yil532 · web-flow · commit a11ded0aca51 · 2023-07-26T06:33:32.000-05:00
* fix: Update default training steps to 300

* remove outdated comment
diff --git a/generative_ai/tuning.py b/generative_ai/tuning.py
@@ -28,7 +28,7 @@ def tuning(
     project_id: str,
     location: str,
     training_data: pd.DataFrame | str,
-    train_steps: int = 10,
+    train_steps: int = 300,
 ) -> None:
     """Tune a new model, based on a prompt-response data.
 
@@ -57,7 +57,7 @@ def tuning(
         training_data=training_data,
         # Optional:
         train_steps=train_steps,
-        tuning_job_location="europe-west4",  # Only supported in europe-west4 for Public Preview
+        tuning_job_location="europe-west4",
         tuned_model_location=location,
     )
 
