diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py b/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py index 936a77b8c..52eef52da 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py @@ -1229,13 +1229,13 @@ class SwooshFunction(torch.autograd.Function): if x.dtype == torch.float16: x = x.to(torch.float32) - one = torch.tensor(1.0, dtype=x.dtype, device=x.device) + zero = torch.tensor(0.0, dtype=x.dtype, device=x.device) with torch.cuda.amp.autocast(enabled=False): with torch.enable_grad(): x = x.detach() x.requires_grad = True - y = torch.logaddexp(one, x - 1.125) - 0.08 * x - 0.3 + y = torch.logaddexp(zero, x - 1.125) - 0.08 * x - 0.3 if not requires_grad: return y