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