Skip to content

Commit

Permalink
docs: fixed a docstring for train_steps
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 547570950
  • Loading branch information
vertex-sdk-bot authored and copybara-github committed Jul 12, 2023
1 parent a4bd9bd commit 1f55b05
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion vertexai/language_models/_language_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -151,7 +151,7 @@ def tune_model(
Args:
training_data: A Pandas DataFrame or a URI pointing to data in JSON lines format.
The dataset must have the "input_text" and "output_text" columns.
train_steps: Number of training steps to perform.
train_steps: Number of training batches to tune on (batch size is 8 samples).
learning_rate: Learning rate for the tuning
tuning_job_location: GCP location where the tuning job should be run. Only "europe-west4" is supported for now.
tuned_model_location: GCP location where the tuned model should be deployed. Only "us-central1" is supported for now.
Expand Down

0 comments on commit 1f55b05

Please sign in to comment.