mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix bug in backprop of random_clamp()

Browse Source
This commit is contained in:
Daniel Povey 2022-10-20 12:49:29 +08:00
parent d37c159174
commit f6b8f0f631
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py
View File

@ -175,7 +175,6 @@ class RandomClampFunction(torch.autograd.Function):
ctx.reflect = reflect
if reflect != 0.0:
ans = ans * (1.0 + reflect) - (x * reflect)
return ans
@staticmethod
@ -185,7 +184,7 @@ class RandomClampFunction(torch.autograd.Function):
reflect = ctx.reflect
if reflect != 0.0:
x_grad = x_grad * (1.0 + reflect) - (ans_grad * reflect)
return ans_grad * is_same.to(ans_grad.dtype), None, None, None, None
return x_grad, None, None, None, None
def random_clamp(x: Tensor,
min: Optional[float] = None,