tf.raw_ops.ResourceApplyCenteredRMSProp
Обновление '*var' в соответствии с алгоритмом centered RMSProp.
tf.raw_ops.ResourceApplyCenteredRMSProp(
var,
mg,
ms,
mom,
lr,
rho,
momentum,
epsilon,
grad,
use_locking=False,
name=None
)
Алгоритм centered RMSProp использует оценку центрального второго момента (т. е. дисперсии) для нормализации, в отличие от обычного RMSProp, который использует (нецентральный) второй момент. Это часто помогает при обучении, но немного дороже в плане вычислений и памяти.
Обратите внимание, что в плотной реализации этого алгоритма mg, ms и mom будут обновляться, даже если градиент равен нулю, но в этой разреженной реализации mg, ms и mom не будут обновляться в итерациях, во время которых градиент равен нулю.
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 | A Tensor типа resource. Должно быть из Variable(). |
mg | A Tensor типа resource. Должно быть из Variable(). |
ms | A Tensor типа resource. Должно быть из Variable(). |
mom | A Tensor типа resource. Должно быть из Variable(). |
lr | A Tensor. Должен быть одного из следующих типов: float32, float64, int32, uint8, int16, int8, complex64, int64, qint8, quint8, qint32, bfloat16, qint16, quint16, uint16, complex128, half, uint32, uint64. Коэффициент масштабирования. Должен быть скаляром. |
rho | A Tensor. Должен иметь тот же тип, что и lr. Скорость затухания. Должен быть скаляром. |
momentum | A Tensor. Должен иметь тот же тип, что и lr. Масштаб импульса. Должен быть скаляром. |
epsilon | A Tensor. Должен иметь тот же тип, что и lr. Слагаемое Риджа. Должен быть скаляром. |
grad | A Tensor. Должен иметь тот же тип, что и lr. Градиент. |
use_locking | Необязательный bool. По умолчанию False. Если True, обновление тензоров var, mg, ms и mom защищено блокировкой; в противном случае поведение не определено, но может характеризоваться меньшей конкуренцией. |
name | Название операции (необязательно). |
| Возвращаемое значение | |
|---|---|
| Созданная операция. |
© 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