tf.compat.v1.tpu.experimental.FtrlParameters
Параметры оптимизации для Ftrl с TPU-вложениями.
tf.compat.v1.tpu.experimental.FtrlParameters(
learning_rate: float,
learning_rate_power: float = -0.5,
initial_accumulator_value: float = 0.1,
l1_regularization_strength: float = 0.0,
l2_regularization_strength: float = 0.0,
use_gradient_accumulation: bool = True,
clip_weight_min: Optional[float] = None,
clip_weight_max: Optional[float] = None,
weight_decay_factor: Optional[float] = None,
multiply_weight_decay_factor_by_learning_rate: Optional[bool] = None,
multiply_linear_by_learning_rate: bool = False,
beta: float = 0,
allow_zero_accumulator: bool = False,
clip_gradient_min: Optional[float] = None,
clip_gradient_max: Optional[float] = None
)
Передайте их в tf.estimator.tpu.experimental.EmbeddingConfigSpec через аргумент optimization_parameters, чтобы установить оптимизатор и его параметры. Подробнее см. документацию по tf.estimator.tpu.experimental.EmbeddingConfigSpec.
estimator = tf.estimator.tpu.TPUEstimator(
...
embedding_config_spec=tf.estimator.tpu.experimental.EmbeddingConfigSpec(
...
optimization_parameters=tf.tpu.experimental.FtrlParameters(0.1),
...))
| Аргументы | |
|---|---|
learning_rate | значение с плавающей точкой. Скорость обучения. |
learning_rate_power | Вещественное значение, которое должно быть меньше или равно нулю. Управляет тем, как скорость обучения уменьшается во время обучения. Используйте ноль для фиксированной скорости обучения. См. раздел 3.1 в статье. |
initial_accumulator_value | Начальное значение для аккумуляторов. Допускаются только нулевые или положительные значения. |
l1_regularization_strength | Вещественное значение, которое должно быть больше или равно нулю. |
l2_regularization_strength | Вещественное значение, которое должно быть больше или равно нулю. |
use_gradient_accumulation | Установка этого значения в False делает вычисление градиентов вложений менее точным, но быстрее. Подробнее см. optimization_parameters.proto. |
clip_weight_min | минимальное значение для обрезки; None означает -бесконечность. |
clip_weight_max | максимальное значение для обрезки; None означает +бесконечность. |
weight_decay_factor | величина распада веса, которую нужно применить; None означает, что веса не распадаются. |
multiply_weight_decay_factor_by_learning_rate | если true, weight_decay_factor умножается на текущую скорость обучения. |
multiply_linear_by_learning_rate | Если значение истинно, то использование линейного слота в обновлении веса умножается на скорость обучения. Это полезно при увеличении скорости обучения от 0 (что обычно приводит к NaN). |
beta | Параметр бета для FTRL. |
allow_zero_accumulator | Изменяет реализацию квадратного корня, чтобы учитывать случай, когда initial_accumulator_value равно нулю. Это приведет к небольшому снижению производительности. |
clip_gradient_min | минимальное значение для обрезки; None означает -бесконечность. |
clip_gradient_max | максимальное значение для обрезки; None означает +бесконечность. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/compat/v1/tpu/experimental/FtrlParameters