Update custom_gradient.py

This commit is contained in:
Vishnuvardhan Janapati 2021-05-10 08:48:30 -07:00 committed by GitHub
parent 6172fb6de0
commit 83739474b5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -203,8 +203,8 @@ def custom_gradient(f=None):
gradient (as calculated by `tf.gradients`) is determined by `f(x)[1]`. gradient (as calculated by `tf.gradients`) is determined by `f(x)[1]`.
Raises: Raises:
RuntimeError: tf.gradients is not supported when eager execution is RuntimeError: `tf.gradients` is not supported when eager execution is
enabled. Use tf.GradientTape instead. enabled. Use `tf.GradientTape` instead.
""" """
if f is None: if f is None: