Remove one line..
This commit is contained in:
parent
0fd2cb141f
commit
c533f91fa2
@ -801,7 +801,6 @@ class DecorrelateFunction(torch.autograd.Function):
|
|||||||
loss.backward()
|
loss.backward()
|
||||||
|
|
||||||
decorr_x_grad = x.grad
|
decorr_x_grad = x.grad
|
||||||
assert x.grad is not None
|
|
||||||
|
|
||||||
# Now, normalize the magnitudes of the rows of the new grad
|
# Now, normalize the magnitudes of the rows of the new grad
|
||||||
# contribution, to have magnitudes equals to ctx.scale times
|
# contribution, to have magnitudes equals to ctx.scale times
|
||||||
|
|||||||
Reference in New Issue
Block a user