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
)
Для главного узла (chief) эта утилита устанавливает правильный инициализатор/восстановитель сессии. Она также создаёт хуки, связанные с сохранением контрольных точек и сводок. Для рабочих узлов (workers) эта утилита устанавливает правильный создатель сессии, который ожидает инициализации/восстановления главного узла. Для получения дополнительной информации см. 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. MetaGraphDefs сохраняются для каждой контрольной точки как model.ckpt-*.meta. |
| Возвращаемое значение | |
|---|---|
Объект MonitoredSession . |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/compat/v1/train/MonitoredTrainingSession