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 несовместим с немедленным выполнением и 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