Hook¶
Описание¶
Hook - базовый класс для семейства вызываемых объектов, которые должны проводить определённые операции над градиентами до обновления весов. Примеры возможных применений хуков: проведение нормировки градиента, обрубание по фиксированному порогу, добавление шума в градиент и т.д.
Создание своего Hook¶
Для реализации кастомизированного хука требуется объявить для него метод __call__:
def call(self, var, state, stream=None):
Параметры
| Параметр | Возможные типы | Описание | По умолчанию |
|---|---|---|---|
| var | Variable | Объект библиотечного класса Variable, являющегося обёрткой вокруг GPU тензоров | - |
| state | str | Строковый индикатор принадлежности var определённому типу данных |
- |
| stream | - | - | None |
Пояснения
-