We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a4bd9bd commit 1f55b05Copy full SHA for 1f55b05
vertexai/language_models/_language_models.py
@@ -151,7 +151,7 @@ def tune_model(
151
Args:
152
training_data: A Pandas DataFrame or a URI pointing to data in JSON lines format.
153
The dataset must have the "input_text" and "output_text" columns.
154
- train_steps: Number of training steps to perform.
+ train_steps: Number of training batches to tune on (batch size is 8 samples).
155
learning_rate: Learning rate for the tuning
156
tuning_job_location: GCP location where the tuning job should be run. Only "europe-west4" is supported for now.
157
tuned_model_location: GCP location where the tuned model should be deployed. Only "us-central1" is supported for now.
0 commit comments