TensorFlow 2.0 RC is available Learn more

tf.contrib.distribute.DistributeConfig

View source on GitHub

Class DistributeConfig

A config tuple for distribution strategies.

Attributes:

  • train_distribute: a DistributionStrategy object for training.
  • eval_distribute: an optional DistributionStrategy object for evaluation.
  • remote_cluster: a dict, ClusterDef or ClusterSpec object specifying the cluster configurations. If this is given, the train_and_evaluate method will be running as a standalone client which connects to the cluster for training.

Properties

train_distribute

eval_distribute

remote_cluster