Spec-Zone.ru › TensorFlow

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 несовместим с eager execution и tf.function. Чтобы перейти на TF2, перепишите код так, чтобы он был совместим с eager execution. Инструкции по замене вызовов Session.run см. в руководстве по переходу. В Keras хуки сессии можно заменить обратными вызовами (Callbacks), например, см. блокнот с примером хука журналирования. Подробнее см. в статье Повышение производительности с помощью tf.function.

Описание

Используется в блокнотах

Используется в руководствах
  • Изучение векторных представлений Swivel CORD-19 из TF-Hub

Для главного узла эта утилита настраивает соответствующий инициализатор/восстановитель сессии. Она также создает хуки для сохранения контрольных точек и сводок. Для рабочих узлов эта утилита настраивает соответствующий создатель сессии, который ожидает инициализации/восстановления главным узлом. Дополнительную информацию см. в разделе 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/api_docs/python/tf/compat/v1/train/MonitoredTrainingSession

Spec-Zone.ru

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