tf.compat.v1.train.SingularMonitoredSession

Объект, подобный сессии, который обрабатывает инициализацию, восстановление и хуки.

tf.compat.v1.train.SingularMonitoredSession(
    hooks=None,
    scaffold=None,
    master='',
    config=None,
    checkpoint_dir=None,
    stop_grace_period_secs=120,
    checkpoint_filename_with_path=None
)

Этот API несовместим с выполнением в режиме eager и tf.function. Для миграции в TF2 перепишите код так, чтобы он был совместим с выполнением в режиме eager. Проверьте руководство по миграции https://www.tensorflow.org/guide/migrate#1_replace_v1sessionrun_calls по замене вызовов Session.run. В Keras хуки сессии можно заменить на Callbacks, например, в блокноте с хуком ведения журнала https://github.com/tensorflow/docs/blob/master/site/en/guide/migrate/logging_stop_hook.ipynb. Для получения более подробной информации, пожалуйста, прочтите https://www.tensorflow.org/guide/function (Лучшая производительность с tf.function).

Описание

Обратите внимание, что это утилита не рекомендуется для распределенных сред. Для распределенных сред, пожалуйста, используйте tf.compat.v1.train.MonitoredSession. Различия между MonitoredSession и SingularMonitoredSession:

• MonitoredSession обрабатывает AbortedError и UnavailableError для распределенных сред, но SingularMonitoredSession этого не делает.
• MonitoredSession может быть создан в режимах chief или worker. SingularMonitoredSession всегда создаётся как chief.
• Вы можете получить доступ к объекту исходной сессии tf.compat.v1.Session, используемой SingularMonitoredSession, в то время как в MonitoredSession исходная сессия является закрытой. Это можно использовать для:
  • Выполнения run без хуков.
  • Сохранения и восстановления.
• Все остальные функции идентичны.

Пример использования:

saver_hook = CheckpointSaverHook(...)
summary_hook = SummarySaverHook(...)
with SingularMonitoredSession(hooks=[saver_hook, summary_hook]) as sess:
  while not sess.should_stop():
    sess.run(train_op)

Инициализация: В момент создания сессия с хуками выполняет следующие действия в указанном порядке:

• вызывает hook.begin() для каждого заданного хука
• завершает граф с помощью scaffold.finalize()
• создаёт сессию
• инициализирует модель с помощью операций инициализации, предоставленных Scaffold
• восстанавливает переменные, если существует контрольная точка
• запускает исполнителей очереди

Выполнение: При вызове run(), сессия с хуками выполняет следующие действия:

• вызывает hook.before_run()
• вызывает TensorFlow session.run() со слиянными запрашиваемыми данными и feed_dict
• вызывает hook.after_run()
• возвращает результат session.run(), запрошенный пользователем

Выход: В момент close() сессия с хуками выполняет следующие действия в указанном порядке:

• вызывает hook.end()
• закрывает исполнителей очереди и сессию
• подавляет ошибку OutOfRange, которая указывает, что все входные данные обработаны, если SingularMonitoredSession используется как контекст.

Подклассы

class StepContext

Методы

close

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

close()

raw_session

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

raw_session()

Возвращает базовый объект TensorFlow.Session.

run

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

run(
    fetches, feed_dict=None, options=None, run_metadata=None
)

Выполнение операций в контролируемой сессии.

Этот метод полностью совместим с методом tf.Session.run().

run_step_fn

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

run_step_fn(
    step_fn
)

Выполнение операций с помощью функции шага.

```python
with tf.Graph().as_default():
  c = tf.compat.v1.placeholder(dtypes.float32)
  v = tf.add(c, 4.0)
  w = tf.add(c, 0.5)
  def step_fn(step_context):
    a = step_context.session.run(fetches=v, feed_dict={c: 0.5})
    if a <= 4.5:
      step_context.request_stop()
      return step_context.run_with_hooks(fetches=w,
                                         feed_dict={c: 0.1})

  with tf.MonitoredSession() as session:
    while not session.should_stop():
      a = session.run_step_fn(step_fn)
```
Hooks interact with the `run_with_hooks()` call inside the
     `step_fn` as they do with a `MonitoredSession.run` call.

should_stop

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

should_stop()

__enter__

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

__enter__()

__exit__

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

__exit__(
    exception_type, exception_value, traceback
)