Spec-Zone.ru › TensorFlow

tf.data.experimental.make_saveable_from_iterator

Возвращает SaveableObject для сохранения/восстановления состояния итератора с помощью Saver. (устарело)

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

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

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

tf.compat.v1.data.experimental.make_saveable_from_iterator

tf.data.experimental.make_saveable_from_iterator(
    iterator, external_state_policy=None
)
Устарело: ЭТА ФУНКЦИЯ УСТАРЕЛА. Она будет удалена в будущей версии. Инструкции по обновлению: make_saveable_from_iterator предназначена для использования в TF1 с tf.compat.v1.Saver. В TF2 используйте tf.train.Checkpoint вместо этого.
Аргументы
iterator Итератор.
external_state_policy Строка, определяющая, как обрабатывать конвейеры ввода, зависящие от внешнего состояния. Возможные значения: 'ignore': Внешнее состояние игнорируется без сообщений. 'warn': Внешнее состояние игнорируется с отображением предупреждения. 'fail': Операция завершается с ошибкой при обнаружении внешнего состояния. По умолчанию установлено значение 'fail'.
Возвращаемое значение
SaveableObject для сохранения/восстановления состояния итератора с помощью Saver.
Исключения
ValueError Если итератор не поддерживает сохранение состояния.
ValueError Если external_state_policy не является одним из 'warn', 'ignore' или 'fail'.

Например:

with tf.Graph().as_default():
  ds = tf.data.Dataset.range(10)
  iterator = ds.make_initializable_iterator()
  # Build the iterator SaveableObject.
  saveable_obj = tf.data.experimental.make_saveable_from_iterator(iterator)
  # Add the SaveableObject to the SAVEABLE_OBJECTS collection so
  # it can be automatically saved using Saver.
  tf.compat.v1.add_to_collection(tf.GraphKeys.SAVEABLE_OBJECTS, saveable_obj)
  saver = tf.compat.v1.train.Saver()

  while continue_training:
    ... Perform training ...
    if should_save_checkpoint:
      saver.save()
Примечание: При восстановлении итератора существующее состояние итератора полностью отбрасывается. Это означает, что любые изменения, которые вы могли внести в граф Dataset, также будут отброшены! Это включает новый граф Dataset, который вы могли построить во время проверки. Поэтому во время выполнения проверки убедитесь, что вы запускаете инициализатор для конвейера ввода проверки после восстановления контрольной точки.
Примечание: Не все итераторы пока поддерживают сохранение состояния. Попытка сохранить состояние недопустимого итератора приведет к ошибке.

© 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/data/experimental/make_saveable_from_iterator

Spec-Zone.ru

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