TensorFlow 1 version View source on GitHub

Returns a SaveableObject for saving/restoring iterator state using Saver.
    iterator, external_state_policy='fail'


  • iterator: Iterator.
  • external_state_policy: A string that identifies how to handle input pipelines that depend on external state. Possible values are 'ignore': The external state is silently ignored. 'warn': The external state is ignored, logging a warning. 'fail': The operation fails upon encountering external state. By default we set it to 'fail'.


A SaveableObject for saving/restoring iterator state using Saver.


  • ValueError: If iterator does not support checkpointing.
  • ValueError: If external_state_policy is not one of 'warn', 'ignore' or 'fail'.

For example:

with tf.Graph().as_default():
  ds =
  iterator = ds.make_initializable_iterator()
  # Build the iterator SaveableObject.
  saveable_obj =
  # 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: