tf.tpu.experimental.AdagradParameters

Параметры оптимизации Adagrad для встраивания TPU.

tf.tpu.experimental.AdagradParameters(
    learning_rate, initial_accumulator=0.1, use_gradient_accumulation=True,
    clip_weight_min=None, clip_weight_max=None
)

Передайте это в tf.estimator.tpu.experimental.EmbeddingConfigSpec через аргумент optimization_parameters, чтобы установить оптимизатор и его параметры. Дополнительные сведения см. в документации для tf.estimator.tpu.experimental.EmbeddingConfigSpec.

estimator = tf.estimator.tpu.TPUEstimator(
    ...
    embedding_spec=tf.estimator.tpu.experimental.EmbeddingConfigSpec(
        ...
        optimization_parameters=tf.tpu.experimental.AdagradParameters(0.1),
        ...))