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, который использует (нецентрированный) второй момент. Это часто помогает в обучении, но немного дороже в вычислительном плане и по памяти.
Обратите внимание, что в плотной реализации этого алгоритма mg, ms и mom будут обновляться, даже если grad равен нулю, но в этой разреженной реализации 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 | Переменная типа resource. Должна быть из Variable(). |
mg | Переменная типа resource. Должна быть из Variable(). |
ms | Переменная типа resource. Должна быть из Variable(). |
mom | Переменная типа resource. Должна быть из Variable(). |
lr | Значение. Должно быть одного из следующих типов: float32, float64, int32, uint8, int16, int8, complex64, int64, qint8, quint8, qint32, bfloat16, uint16, complex128, half, uint32, uint64. Коэффициент масштабирования. Должен быть скаляром. |
rho | Значение. Должно иметь тот же тип, что и lr. Скорость затухания. Должен быть скаляром. |
momentum | Значение. Должно иметь тот же тип, что и lr. Масштаб импульса. Должен быть скаляром. |
epsilon | Значение. Должно иметь тот же тип, что и lr. Слагаемое ridge. Должен быть скаляром. |
grad | Значение. Должно иметь тот же тип, что и lr. Градиент. |
use_locking | Необязательный параметр. По умолчанию 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/versions/r2.9/api_docs/python/tf/raw_ops/ResourceApplyCenteredRMSProp