Spec-Zone.ru › TensorFlow 2.9

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=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/versions/r2.9/api_docs/python/tf/data/experimental/make_saveable_from_iterator

Spec-Zone.ru

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