github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-19 13:09:56 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

_

Browse source
This commit is contained in:
Colombo 2020-03-19 13:29:33 +04:00
commit 4c24f9d41c
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
core/leras/ops/__init__.py
View file

@ -56,7 +56,7 @@ def tf_gradients ( loss, vars ):
gv = [*zip(grads,vars)] gv = [*zip(grads,vars)]
for g,v in gv: for g,v in gv:
if g is None: if g is None:
raise Exception(f"No gradient for variable {v.name}") raise Exception(f"Variable {v.name} is declared as trainable, but no tensors flow through it.")
return gv return gv
nn.gradients = tf_gradients nn.gradients = tf_gradients