Missed TensorFlow World? Check out the recap. Learn more

Module: tf.compat.v2.train

TensorFlow 2.0 version

Support for training models.

See the Training guide.


experimental module: Public API for tf.train.experimental namespace.


class BytesList

class Checkpoint: Groups trackable objects, saving and restoring them.

class CheckpointManager: Deletes old checkpoints.

class ClusterDef

class ClusterSpec: Represents a cluster as a set of "tasks", organized into "jobs".

class Coordinator: A coordinator for threads.

class Example

class ExponentialMovingAverage: Maintains moving averages of variables by employing an exponential decay.

class Feature

class FeatureList

class FeatureLists

class Features

class FloatList

class Int64List

class JobDef

class SequenceExample

class ServerDef


checkpoints_iterator(...): Continuously yield new checkpoint files as they appear.

get_checkpoint_state(...): Returns CheckpointState proto from the "checkpoint" file.

latest_checkpoint(...): Finds the filename of latest saved checkpoint file.

list_variables(...): Returns list of all variables in the checkpoint.

load_checkpoint(...): Returns CheckpointReader for checkpoint found in ckpt_dir_or_file.

load_variable(...): Returns the tensor value of the given variable in the checkpoint.