Browse Source

add debug code.

master
wogong 7 years ago
parent
commit
2d906af47c
  1. 2
      models/functions.py

2
models/functions.py

@ -12,7 +12,7 @@ class ReverseLayerF(Function):
@staticmethod
def backward(ctx, grad_output):
output = grad_output.neg() * ctx.alpha
#print("reverse gradient is {}".format(output))
return output, None

Loading…
Cancel
Save