tf.raw_ops.ResourceApplyCenteredRMSProp
Обновляет '*var' в соответствии с алгоритмом центрированного RMSProp.
tf.raw_ops.ResourceApplyCenteredRMSProp(
var,
mg,
ms,
mom,
lr,
rho,
momentum,
epsilon,
grad,
use_locking=False,
name=None
)
Алгоритм центрированного RMSProp использует оценку центрированного второго момента (т. е. дисперсии) для нормализации, в отличие от обычного RMSProp, который использует (нецентрированный) второй момент. Это часто помогает в обучении, но требует немного больше вычислительных ресурсов и памяти.
Обратите внимание, что в плотной (dense) реализации этого алгоритма mg, ms и mom будут обновляться, даже если grad равен нулю, но в этой разреженной (sparse) реализации mg, ms и mom не будут обновляться на итерациях, где grad равен нулю.
mean_square = decay * mean_square + (1-decay) * gradient ** 2 mean_grad = decay * mean_grad + (1-decay) * gradient
Delta = learning_rate * gradient / sqrt(mean_square + epsilon - mean_grad ** 2)
mg <- rho * mg{t-1} + (1-rho) * grad ms <- rho * ms{t-1} + (1-rho) * grad * grad mom <- momentum * mom_{t-1} + lr * grad / sqrt(ms - mg * mg + epsilon) var <- var - mom
| Аргументы | |
|---|---|
var | Tensor типа resource. Должен быть получен из Variable(). |
mg | Tensor типа resource. Должен быть получен из Variable(). |
ms | Tensor типа resource. Должен быть получен из Variable(). |
mom | Tensor типа resource. Должен быть получен из Variable(). |
lr | Tensor. Должен быть одного из следующих типов: float32, float64, int32, uint8, int16, int8, complex64, int64, qint8, quint8, qint32, bfloat16, qint16, quint16, uint16, complex128, half, uint32, uint64. Масштабирующий коэффициент. Должен быть скаляром. |
rho | Tensor. Должен иметь тот же тип, что и lr. Скорость затухания. Должен быть скаляром. |
momentum | Tensor. Должен иметь тот же тип, что и lr. Масштаб импульса (momentum). Должен быть скаляром. |
epsilon | Tensor. Должен иметь тот же тип, что и lr. Гребневый член (ridge term). Должен быть скаляром. |
grad | Tensor. Должен иметь тот же тип, что и lr. Градиент. |
use_locking | Необязательный bool. По умолчанию False. Если True, обновление тензоров var, mg, ms и mom защищено блокировкой; в противном случае поведение не определено, но может наблюдаться меньшая состязательность (contention). |
name | Имя для операции (необязательно). |
| Возвращает | |
|---|---|
| Созданный Operation. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/raw_ops/ResourceApplyCenteredRMSProp