withtf.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()whilecontinue_training:...Performtraining...ifshould_save_checkpoint:saver.save()