Type StochasticGradientDescentParameters
Namespace tensorflow.tpu.experimental
Parent _OptimizationParameters
Interfaces IStochasticGradientDescentParameters
Optimization parameters for stochastic gradient descent for TPU embeddings. Pass this to
tf.estimator.tpu.experimental.EmbeddingConfigSpec
via the
`optimization_parameters` argument to set the optimizer and its parameters.
See the documentation for tf.estimator.tpu.experimental.EmbeddingConfigSpec
for more details. ```
estimator = tf.estimator.tpu.TPUEstimator(
...
embedding_config_spec=tf.estimator.tpu.experimental.EmbeddingConfigSpec(
...
optimization_parameters=(
tf.tpu.experimental.StochasticGradientDescentParameters(0.1))))
```