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.3/api_docs/python/tf/compat/v1/train/MonitoredTrainingSession