tf.compat.v1.train.FeedFnHook

Выполняет feed_fn и устанавливает feed_dict соответственно.

Наследуется от: SessionRunHook

tf.compat.v1.train.FeedFnHook(
    feed_fn
)

Методы

after_create_session

Просмотреть исходный код

after_create_session(
    session, coord
)

Вызывается при создании новой сессии TensorFlow.

Этот метод вызывается, чтобы сообщить хукам о создании новой сессии. Это отличается от ситуации, когда вызывается begin, в двух ключевых аспектах:

• При вызове этого метода, граф уже финализирован и больше невозможно добавлять в него операции.
• Этот метод также вызывается при восстановлении обернутой сессии, а не только в начале общей сессии.

after_run

Просмотреть исходный код

after_run(
    run_context, run_values
)

Вызывается после каждого вызова run().

Аргумент run_values содержит результаты запрошенных операций/тензоров функцией before_run().

Аргумент run_context — тот же, что и переданный в вызов before_run. Вы можете вызвать run_context.request_stop(), чтобы остановить итерацию.

Если session.run() вызывает какие-либо исключения, то after_run() не вызывается.

before_run

Просмотреть исходный код

before_run(
    run_context
)

Вызывается перед каждым вызовом run().

Вы можете вернуть из этого вызова объект SessionRunArgs, указывающий операции или тензоры для добавления в следующий вызов run(). Эти операции/тензоры будут выполнены вместе с исходно переданными в первоначальный вызов run() операциями/тензорами. Возвращаемые вами аргументы для run также могут содержать фиды для добавления в вызов run().

Аргумент run_context — это объект SessionRunContext, который предоставляет информацию о предстоящем вызове run(): исходно запрошенные операции/тензоры и сессия TensorFlow.

На этом этапе граф финализирован, и вы больше не можете добавлять операции.

begin

Просмотреть исходный код

begin()

Вызывается один раз перед использованием сессии.

При вызове, текущий граф будет запущен в сессии. Хук может изменять граф, добавляя новые операции. После вызова begin(), граф финализируется, и другие колбэки больше не могут его изменять. Второй вызов begin() для того же графа не должен изменять граф.

end

Просмотреть исходный код

end(
    session
)

Вызывается в конце сессии.

Аргумент session можно использовать, если хук хочет выполнить финальные операции, такие как сохранение последнего контрольного пункта.

Если session.run() вызывает исключение, отличное от OutOfRangeError или StopIteration, то end() не вызывается. Обратите внимание на разницу в поведении end() и after_run(), когда session.run() вызывает OutOfRangeError или StopIteration. В этом случае end() вызывается, но after_run() — нет.