tf.compat.v1.train.SessionManager

Помощник для обучения, который восстанавливает модель из контрольной точки и создаёт сессию.

tf.compat.v1.train.SessionManager(
    local_init_op=None,
    ready_op=None,
    ready_for_local_init_op=None,
    graph=None,
    recovery_wait_secs=30,
    local_init_run_options=None,
    local_init_feed_dict=None
)

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

• Сохранение обученных переменных по мере обучения.
• Инициализация переменных при запуске, восстановление их из последней контрольной точки после сбоя или ожидание появления контрольных точек.

Использование:

with tf.Graph().as_default():
   ...add operations to the graph...
  # Create a SessionManager that will checkpoint the model in '/tmp/mydir'.
  sm = SessionManager()
  sess = sm.prepare_session(master, init_op, saver, checkpoint_dir)
  # Use the session to train the graph.
  while True:
    sess.run(<my_train_op>)

prepare_session() инициализирует или восстанавливает модель. Требует init_op и saver в качестве аргумента.

Второй процесс может ожидать готовности модели, выполнив следующие действия:

with tf.Graph().as_default():
   ...add operations to the graph...
  # Create a SessionManager that will wait for the model to become ready.
  sm = SessionManager()
  sess = sm.wait_for_session(master)
  # Use the session to train the graph.
  while True:
    sess.run(<my_train_op>)

wait_for_session() ожидает инициализации модели другими процессами.

Методы

prepare_session

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

prepare_session(
    master,
    init_op=None,
    saver=None,
    checkpoint_dir=None,
    checkpoint_filename_with_path=None,
    wait_for_checkpoint=False,
    max_wait_secs=7200,
    config=None,
    init_feed_dict=None,
    init_fn=None
)

Создаёт Session. Убеждается, что модель готова к использованию.

Создаёт Session на 'master'. Если передан объект saver, и checkpoint_dir указывает на директорию с допустимыми файлами контрольных точек, то попытается восстановить модель из контрольной точки. Если файлы контрольных точек отсутствуют, и wait_for_checkpoint имеет значение True, процесс будет проверять каждые recovery_wait_secs, до max_wait_secs, пока восстановление не произойдёт.

Если модель не может быть восстановлена, она инициализируется путём выполнения init_op и вызова init_fn, если они предоставлены. local_init_op также выполняется после init_op и init_fn, независимо от успешного восстановления модели, но только если ready_for_local_init_op пройдёт.

Если модель восстанавливается из контрольной точки, предполагается, что все глобальные переменные были инициализированы, в частности, ни init_op, ни init_fn не будут выполнены.

Ошибка, если модель не может быть восстановлена и нет init_op или init_fn или local_init_op.

recover_session

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

recover_session(
    master,
    saver=None,
    checkpoint_dir=None,
    checkpoint_filename_with_path=None,
    wait_for_checkpoint=False,
    max_wait_secs=7200,
    config=None
)

Создаёт Session, если возможно, восстанавливая её.

Создаёт новую сессию на 'master'. Если сессия не инициализирована и может быть восстановлена из контрольной точки, восстанавливает её.

wait_for_session

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

wait_for_session(
    master, config=None, max_wait_secs=float('Inf')
)

Создаёт новую Session и ожидает готовности модели.

Создаёт новую Session на 'master'. Ожидает инициализации модели или восстановления из контрольной точки. Ожидается, что другой поток или процесс сделают модель готовой, и это предназначено для потоков/процессов, участвующих в конфигурации распределённого обучения, где другой поток/процесс отвечает за инициализацию или восстановление обучаемой модели.

Примечание: Время ожидания этим методом сессии ограничено max_wait_secs. По умолчанию эта функция будет ожидать бесконечно.