Spec-Zone.ru › TensorFlow 1.15

tf.train.checkpoints_iterator

Просмотреть исходный код на GitHub

Непрерывно возвращает новые файлы контрольных точек по мере их появления.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительных сведений.

tf.compat.v1.train.checkpoints_iterator, `tf.compat.v2.train.checkpoints_iterator`

tf.train.checkpoints_iterator(
    checkpoint_dir, min_interval_secs=0, timeout=None, timeout_fn=None
)

Итератор проверяет наличие новых контрольных точек только тогда, когда поток управления вернулся к нему. Это означает, что он может пропустить контрольные точки, если выполнение вашего кода между итерациями занимает больше времени, чем min_interval_secs или интервал, в который записываются новые контрольные точки.

Аргумент timeout — максимальное количество секунд ожидания новой контрольной точки. Он используется в сочетании с аргументом timeout_fn, следующим образом:

  • Если таймаут истекает, и timeout_fn не было указано, итератор перестаёт возвращать значения.
  • Если timeout_fn было указано, эта функция вызывается, и если она возвращает истинное значение булевого типа, итератор прекращает возвращать значения.
  • Если функция возвращает ложное значение булевого типа, итератор возобновляет ожидание новых контрольных точек. В этот момент снова применяется логика таймаута.

Это поведение позволяет вызывающим сторонам контролировать, что делать, если контрольные точки не генерируются достаточно быстро или перестают генерироваться. Например, если вызывающие стороны могут определить, что обучение остановлено и известно, что новые контрольные точки не будут генерироваться, они могут предоставить timeout_fn , которая возвращает True, когда обучение остановлено. Если они знают, что обучение продолжается, они возвращают False вместо этого.

Аргументы
checkpoint_dir Каталог, в котором сохраняются контрольные точки.
min_interval_secs Минимальное количество секунд между возвратом контрольных точек.
timeout Максимальное количество секунд ожидания между контрольными точками. Если оставить значение по умолчанию, процесс будет ждать неопределённо долго.
timeout_fn Дополнительная функция, вызываемая после истечения таймаута. Если функция возвращает True, это означает, что новые контрольные точки больше не будут генерироваться, и итератор завершится. Функция вызывается без аргументов.

Возвращаемые значения:

Пути к файлам последних контрольных точек по мере их поступления.

© 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/r1.15/api_docs/python/tf/train/checkpoints_iterator

Spec-Zone.ru

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