experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #26142 from danaugrs:patch2

Browse Source 
PiperOrigin-RevId: 237336877
This commit is contained in:
TensorFlower Gardener 2019-03-07 15:27:47 -08:00
commit 90b561ef09
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/keras/optimizer_v2/optimizer_v2.py
View File

@ -358,7 +358,8 @@ class OptimizerV2(trackable.Trackable):
ValueError: In case any gradient cannot be computed (e.g. if gradient ValueError: In case any gradient cannot be computed (e.g. if gradient
function not implemented). function not implemented).
""" """
grads = gradients.gradients(loss, params) with backend.get_graph().as_default():
grads = gradients.gradients(loss, params)
if None in grads: if None in grads:
raise ValueError("An operation has `None` for gradient. " raise ValueError("An operation has `None` for gradient. "
"Please make sure that all of your ops have a " "Please make sure that all of your ops have a "