Spec-Zone.ru › TensorFlow 2.9

tf.compat.v1.train.MonitoredTrainingSession

Создаёт MonitoredSession для обучения.

tf.compat.v1.train.MonitoredTrainingSession(
    master='',
    is_chief=True,
    checkpoint_dir=None,
    scaffold=None,
    hooks=None,
    chief_only_hooks=None,
    save_checkpoint_secs=USE_DEFAULT,
    save_summaries_steps=USE_DEFAULT,
    save_summaries_secs=USE_DEFAULT,
    config=None,
    stop_grace_period_secs=120,
    log_step_count_steps=100,
    max_wait_secs=7200,
    save_checkpoint_steps=USE_DEFAULT,
    summary_dir=None,
    save_graph_def=True
)

Переход на TF2

Внимание: Этот API был разработан для TensorFlow v1. Продолжайте чтение для получения подробностей о том, как перейти от этого API к эквиваленту в чистом TensorFlow v2. Обратитесь к руководству по миграции TensorFlow v1 в TensorFlow v2 для получения инструкций о том, как мигрировать остальную часть вашего кода.

Этот API несовместим с немедленным выполнением и tf.function. Для перехода на TF2 перепишите код, чтобы он был совместим с немедленным выполнением. Проверьте руководство по миграции по замене Session.run вызовов. В Keras, хуки сессии могут быть заменены обратными вызовами, например, блокнот хука ведения журнала. Для получения более подробной информации, пожалуйста, прочитайте Улучшенная производительность с tf.function.

Описание

Для главного узла эта утилита устанавливает правильный инициализатор/восстановитель сессии. Она также создаёт хуки, связанные с сохранением контрольных точек и сводок. Для рабочих узлов эта утилита устанавливает правильный создатель сессии, который ожидает инициализации/восстановления главного узла. Пожалуйста, проверьте tf.compat.v1.train.MonitoredSession для получения дополнительной информации.

Аргументы
master String используемый главный узел TensorFlow.
is_chief Если True, он будет обрабатывать инициализацию и восстановление основной сессии TensorFlow. Если False, он будет ожидать инициализации или восстановления сессии TensorFlow главным узлом.
checkpoint_dir Строка. Необязательный путь к каталогу для восстановления переменных.
scaffold Scaffold для сбора или построения поддерживающих операций. Если не указано, создаётся значение по умолчанию. Используется для завершения графа.
hooks Необязательный список объектов SessionRunHook.
chief_only_hooks список объектов SessionRunHook. Активировать эти хуки, если is_chief==True, в противном случае игнорировать.
save_checkpoint_secs Частота, в секундах, с которой контрольная точка сохраняется с помощью сохранённого объекта контрольной точки по умолчанию. Если оба save_checkpoint_steps и save_checkpoint_secs установлены в None, тогда сохранённый объект контрольной точки по умолчанию не используется. Если оба указаны, используется только save_checkpoint_secs. Значение по умолчанию 600.
save_summaries_steps Частота, в количестве глобальных шагов, с которой сводки записываются на диск с помощью сохранённого объекта сводки по умолчанию. Если оба save_summaries_steps и save_summaries_secs установлены в None, тогда сохранённый объект сводки по умолчанию не используется. Значение по умолчанию 100.
save_summaries_secs Частота, в секундах, с которой сводки записываются на диск с помощью сохранённого объекта сводки по умолчанию. Если оба save_summaries_steps и save_summaries_secs установлены в None, тогда сохранённый объект сводки по умолчанию не используется. Значение по умолчанию — отключено.
config экземпляр tf.compat.v1.ConfigProto протокола, используемого для настройки сессии. Это config аргумент конструктора tf.compat.v1.Session.
stop_grace_period_secs Количество секунд, отводимых потокам на остановку после вызова close().
log_step_count_steps Частота, в количестве глобальных шагов, с которой регистрируется глобальный шаг/секунда.
max_wait_secs Максимальное время ожидания рабочими узлами доступности сессии. Это должно быть относительно коротким, чтобы помочь обнаружить неправильный код, но иногда может потребоваться увеличить, если главный узел долго запускается.
save_checkpoint_steps Частота, в количестве глобальных шагов, с которой контрольная точка сохраняется с помощью сохранённого объекта контрольной точки по умолчанию. Если оба save_checkpoint_steps и save_checkpoint_secs установлены в None, тогда сохранённый объект контрольной точки по умолчанию не используется. Если оба указаны, используется только save_checkpoint_secs. Значение по умолчанию — отключено.
summary_dir Строка. Необязательный путь к каталогу для сохранения сводок. Если None, используется checkpoint_dir.
save_graph_def Нужно ли сохранять GraphDef и MetaGraphDef в checkpoint_dir. GraphDef сохраняется после создания сессии как graph.pbtxt. MetaGraphDef сохраняются для каждой контрольной точки как model.ckpt-*.meta.
Возвращает
Объект 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/compat/v1/train/MonitoredTrainingSession

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API