Spec-Zone.ru › TensorFlow 2.4

tf.data.experimental.make_saveable_from_iterator

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

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

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

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

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

tf.compat.v1.data.experimental.make_saveable_from_iterator

tf.data.experimental.make_saveable_from_iterator(
    iterator, external_state_policy='fail'
)
Предупреждение: ЭТА ФУНКЦИЯ УСТАРЕЛА. Она будет удалена в будущей версии. Инструкции по обновлению: 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, который вы могли построить во время проверки. Поэтому, во время выполнения проверки убедитесь, что вы запускаете инициализатор для конвейера входных данных проверки после восстановления контрольной точки.
Примечание: Не все итераторы пока поддерживают сохранение точки останова. Попытка сохранить состояние недопустимого итератора приведет к ошибке.

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

Spec-Zone.ru

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