diff --git a/tensorflow/python/data/experimental/ops/iterator_ops.py b/tensorflow/python/data/experimental/ops/iterator_ops.py index e8b72bc25bb..7a95e32a32e 100644 --- a/tensorflow/python/data/experimental/ops/iterator_ops.py +++ b/tensorflow/python/data/experimental/ops/iterator_ops.py @@ -25,6 +25,7 @@ from tensorflow.python.training import basic_session_run_hooks from tensorflow.python.training import checkpoint_management from tensorflow.python.training import saver as saver_lib from tensorflow.python.training import session_run_hook +from tensorflow.python.util import deprecation from tensorflow.python.util.tf_export import tf_export @@ -42,6 +43,9 @@ def _convert_external_state_policy_to_enum(external_state_policy): @tf_export("data.experimental.make_saveable_from_iterator") +@deprecation.deprecated( + None, "`make_saveable_from_iterator` is intended for use in TF1 with " + "`tf.compat.v1.Saver`. In TF2, use `tf.train.Checkpoint` instead.") def make_saveable_from_iterator(iterator, external_state_policy="fail"): """Returns a SaveableObject for saving/restoring iterator state using Saver. diff --git a/tensorflow/python/training/tracking/util.py b/tensorflow/python/training/tracking/util.py index 144ec068551..0814297d0d1 100644 --- a/tensorflow/python/training/tracking/util.py +++ b/tensorflow/python/training/tracking/util.py @@ -1723,9 +1723,9 @@ class Checkpoint(tracking.AutoTrackable): `Checkpoint`'s constructor accepts keyword arguments whose values are types that contain trackable state, such as `tf.keras.optimizers.Optimizer` - implementations, `tf.Variable`, `tf.keras.Layer` implementations, or - `tf.keras.Model` implementations. It saves these values with a checkpoint, and - maintains a `save_counter` for numbering checkpoints. + implementations, `tf.Variable`s, `tf.data.Dataset` iterators, `tf.keras.Layer` + implementations, or `tf.keras.Model` implementations. It saves these values + with a checkpoint, and maintains a `save_counter` for numbering checkpoints. Example usage: