google.ai.generativelanguage.Hyperparameters

Hyperparameters controlling the tuning process.

epoch_count int

Immutable. The number of training epochs. An epoch is one pass through the training data. If not set, a default of 10 will be used.

batch_size int

Immutable. The batch size hyperparameter for tuning. If not set, a default of 16 or 64 will be used based on the number of training examples.

learning_rate float

Immutable. The learning rate hyperparameter for tuning. If not set, a default of 0.0002 or 0.002 will be calculated based on the number of training examples.