Internal change

PiperOrigin-RevId: 336215069

Internal change
PiperOrigin-RevId: 336215069
dc2ae42c · Ran Chen · A. Unique TensorFlower · 2fa75516 · dc2ae42c
Commit dc2ae42c authored Oct 08, 2020 by Ran Chen Committed by A. Unique TensorFlower Oct 08, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/staging/training/grad_utils.py official/staging/training/grad_utils.py +1 -1

No files found.
--- a/official/staging/training/grad_utils.py
+++ b/official/staging/training/grad_utils.py
@@ -73,7 +73,7 @@ def _filter_and_allreduce_gradients(grads_and_vars,
  hints = tf.distribute.experimental.CollectiveHints(
      bytes_per_pack=bytes_per_pack)
  allreduced_grads = tf.distribute.get_replica_context().all_reduce(
-      tf.distribute.ReduceOp.SUM, grads, experimental_hints=hints)
+      tf.distribute.ReduceOp.SUM, grads, hints)
  if allreduce_precision == "float16":
    allreduced_grads = [tf.cast(grad, "float32") for grad in allreduced_grads]
  return allreduced_grads, variables