tf.estimator.SessionRunContext
Предоставляет информацию о выполняемом session.run() вызове.
tf.estimator.SessionRunContext(
original_args, session
)
Предоставляет информацию об исходном запросе к функции Session.Run(). Объекты SessionRunHook могут остановить цикл, вызвав request_stop() объекта run_context. В будущем мы можем использовать этот объект для добавления дополнительной информации о запуске без изменения API хуков.
| Атрибуты | |
|---|---|
original_args | Объект SessionRunArgs, содержащий исходные аргументы функции run(). Если пользователь вызвал |
session | Объект сессии TensorFlow, который выполнит run. |
stop_requested | Возвращает, запрошен ли стоп. Если true, то |
Методы
request_stop
request_stop()
Устанавливает запрос на остановку.
Хуки могут использовать эту функцию для запроса остановки итераций. MonitoredSession проверяет, был ли этот вызов сделан.
© 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/estimator/SessionRunContext