Spec-Zone.ru › TensorFlow 2.3

tf.data.experimental.make_saveable_from_iterator

View source on 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.3/api_docs/python/tf/data/experimental/make_saveable_from_iterator

Spec-Zone.ru

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