tf.compat.v1.train.MonitoredSession.StepContext
Инструмент управления потоком для step_fn из run_step_fn().
tf.compat.v1.train.MonitoredSession.StepContext(
session, run_with_hooks_fn
)
Пользователи step_fn могут выполнять run() вызовы без запуска хуков, получив доступ к session. Вызов run() с хуками можно выполнить, используя run_with_hooks(). Поток вычислений можно прервать, используя request_stop().
| Аргументы | |
|---|---|
session | Экземпляр tf.compat.v1.Session. |
run_with_hooks_fn | Функция для выполнения извлечений и хуков. |
| Атрибуты | |
|---|---|
session | |
Методы
request_stop
request_stop()
Завершите цикл обучения, вызвав should_stop() для возврата True.
Заставляет step_fn завершиться, вызвав исключение.
| Исключения | |
|---|---|
| StopIteration |
run_with_hooks
run_with_hooks(
*args, **kwargs
)
То же, что и MonitoredSession.run. Принимает те же аргументы.
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/compat/v1/train/MonitoredSession/StepContext