google.generativeai.protos.Hyperparameters
Hyperparameters controlling the tuning process.
Read more at
https://ai.google.dev/docs/model_tuning_guidance
This message has oneof_ fields (mutually exclusive fields).
For each oneof, at most one member field can be set at the same time.
Setting any member of the oneof automatically clears all other
members.
Attributes |
|---|
|
`learning_rate`
|
`float`
Optional. Immutable. The learning rate
hyperparameter for tuning. If not set, a default
of 0.001 or 0.0002 will be calculated based on
the number of training examples.
This field is a member of `oneof`_ ``learning_rate_option``.
|
|
`learning_rate_multiplier`
|
`float`
Optional. Immutable. The learning rate multiplier is used to
calculate a final learning_rate based on the default
(recommended) value. Actual learning rate :=
learning_rate_multiplier \* default learning rate Default
learning rate is dependent on base model and dataset size.
If not set, a default of 1.0 will be used.
This field is a member of `oneof`_ ``learning_rate_option``.
|
|
`epoch_count`
|
`int`
Immutable. The number of training epochs. An
epoch is one pass through the training data. If
not set, a default of 5 will be used.
|
|
`batch_size`
|
`int`
Immutable. The batch size hyperparameter for
tuning. If not set, a default of 4 or 16 will be
used based on the number of training examples.
|
View source on GitHub