Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf_agents.utils.eager_utils.add_gradients_summaries

View source on GitHub

Add summaries to gradients.

tf_agents.utils.eager_utils.add_gradients_summaries(
    grads_and_vars, step
)

Args:

  • grads_and_vars: A list of gradient to variable pairs (tuples).
  • step: Variable to use for summaries.