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. Принимает те же аргументы.
© 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/compat/v1/train/MonitoredSession/StepContext