diff options
Diffstat (limited to 'tensorflow/python/training/training.py')
-rw-r--r-- | tensorflow/python/training/training.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/training/training.py b/tensorflow/python/training/training.py index a814eb99ce..284cc43bc4 100644 --- a/tensorflow/python/training/training.py +++ b/tensorflow/python/training/training.py @@ -182,6 +182,7 @@ from tensorflow.python.training.rmsprop import RMSPropOptimizer from tensorflow.python.training.gradient_descent import GradientDescentOptimizer from tensorflow.python.training.proximal_gradient_descent import ProximalGradientDescentOptimizer from tensorflow.python.training.sync_replicas_optimizer import SyncReplicasOptimizer +from tensorflow.python.training.sync_replicas_optimizer import SyncReplicasOptimizerV2 # Utility classes for training. from tensorflow.python.training.coordinator import Coordinator |