Обновите «*var» в соответствии с алгоритмом Адама.
$$\text{lr}_t := \mathrm{learning_rate} * \sqrt{1 - \beta_2^t} / (1 - \beta_1^t)$$$$m_t := \beta_1 * m_{t-1} + (1 - \beta_1) * g$$$$v_t := \beta_2 * v_{t-1} + (1 - \beta_2) * g * g$$$$\text{variable} := \text{variable} - \text{lr}_t * m_t / (\sqrt{v_t} + \epsilon)$$
Вложенные классы
сорт | РесурсПрименитьАдам.Параметры | Дополнительные атрибуты для ResourceApplyAdam |
Константы
Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
статический <T расширяет TType > ResourceApplyAdam | create ( Область видимости, Операнд <?> var, Операнд <?> m, Операнд <?> v, Операнд <T> beta1Power, Операнд <T> beta2Power, Операнд <T> lr, Операнд <T> beta1, Операнд <T > beta2, Операнд <T> эпсилон, Операнд <T> град, Опции... опции) Фабричный метод для создания класса, обертывающего новую операцию ResourceApplyAdam. |
статический ResourceApplyAdam.Options | useLocking (логическое значение useLocking) |
статический ResourceApplyAdam.Options | useNesterov (логическое значение useNesterov) |
Унаследованные методы
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Публичные методы
public static ResourceApplyAdam create ( Область действия , Операнд <?> var, Операнд <?> m, Операнд <?> v, Операнд <T> beta1Power, Операнд <T> beta2Power, Операнд <T> lr, Операнд <T> beta1, Операнд <T> бета2, Операнд <T> эпсилон, Операнд <T> град, Опции... опции)
Фабричный метод для создания класса, обертывающего новую операцию ResourceApplyAdam.
Параметры
объем | текущий объем |
---|---|
вар | Должно быть из переменной(). |
м | Должно быть из переменной(). |
в | Должно быть из переменной(). |
бета1Power | Должно быть скаляр. |
бета2Power | Должно быть скаляр. |
лр | Масштабирующий коэффициент. Должно быть скаляр. |
бета1 | Фактор импульса. Должно быть скаляр. |
бета2 | Фактор импульса. Должно быть скаляр. |
эпсилон | Риджовый термин. Должно быть скаляр. |
выпускник | Градиент. |
параметры | содержит значения необязательных атрибутов |
Возврат
- новый экземпляр ResourceApplyAdam
public static ResourceApplyAdam.Options useLocking (логическое значение useLocking)
Параметры
использоватьLocking | Если «True», обновление тензоров var, m и v будет защищено блокировкой; в противном случае поведение не определено, но может вызывать меньше конфликтов. |
---|
public static ResourceApplyAdam.Options useNesterov (логическое значение useNesterov)
Параметры
использовать Нестерова | Если «True», используется обновление нестеров. |
---|