diff options
Diffstat (limited to 'tensorflow/contrib/distribute/python/examples/simple_tfkeras_example.py')
-rw-r--r-- | tensorflow/contrib/distribute/python/examples/simple_tfkeras_example.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/distribute/python/examples/simple_tfkeras_example.py b/tensorflow/contrib/distribute/python/examples/simple_tfkeras_example.py index e714255f69..b87224251c 100644 --- a/tensorflow/contrib/distribute/python/examples/simple_tfkeras_example.py +++ b/tensorflow/contrib/distribute/python/examples/simple_tfkeras_example.py @@ -41,7 +41,7 @@ def main(args): strategy = tf.contrib.distribute.MirroredStrategy( ['/device:GPU:0', '/device:GPU:1']) - config = tf.estimator.RunConfig(distribute=strategy) + config = tf.estimator.RunConfig(train_distribute=strategy) optimizer = tf.train.GradientDescentOptimizer(0.2) model = tf.keras.Sequential() |