Skip to content

Commit 1f55b05

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
docs: fixed a docstring for train_steps
PiperOrigin-RevId: 547570950
1 parent a4bd9bd commit 1f55b05

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vertexai/language_models/_language_models.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,7 @@ def tune_model(
151151
Args:
152152
training_data: A Pandas DataFrame or a URI pointing to data in JSON lines format.
153153
The dataset must have the "input_text" and "output_text" columns.
154-
train_steps: Number of training steps to perform.
154+
train_steps: Number of training batches to tune on (batch size is 8 samples).
155155
learning_rate: Learning rate for the tuning
156156
tuning_job_location: GCP location where the tuning job should be run. Only "europe-west4" is supported for now.
157157
tuned_model_location: GCP location where the tuned model should be deployed. Only "us-central1" is supported for now.

0 commit comments

Comments
 (0)