mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 01:52:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix backprop bug

Browse Source
This commit is contained in:
Daniel Povey 2022-03-04 12:29:44 +08:00
parent cd216f50b6
commit 3d9ddc2016
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/conformer_ctc/subsampling.py
View File

@ -220,7 +220,7 @@ def _exp_scale_swish_backward(x: Tensor, scale: Tensor, speed: float) -> Tensor:
class ExpScaleSwishFunction(torch.autograd.Function): class ExpScaleSwishFunction(torch.autograd.Function):
@staticmethod @staticmethod
def forward(ctx, x: Tensor, scale: Tensor, speed: float) -> Tensor: def forward(ctx, x: Tensor, scale: Tensor, speed: float) -> Tensor:
ctx.save_for_backward(x, scale) ctx.save_for_backward(x.detach(), scale.detach())
ctx.speed = speed ctx.speed = speed
return _exp_scale_swish(x, scale, speed) return _exp_scale_swish(x, scale, speed)