Hook

Описание

Hook - базовый класс для семейства вызываемых объектов, которые должны проводить определённые операции над градиентами до обновления весов. Примеры возможных применений хуков: проведение нормировки градиента, обрубание по фиксированному порогу, добавление шума в градиент и т.д.

Создание своего Hook

Для реализации кастомизированного хука требуется объявить для него метод __call__:

def call(self, var, state, stream=None):

Параметры

Параметр Возможные типы Описание По умолчанию
var Variable Объект библиотечного класса Variable, являющегося обёрткой вокруг GPU тензоров -
state str Строковый индикатор принадлежности var определённому типу данных -
stream - - None

Пояснения

-