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 inf issue

Browse Source
This commit is contained in:
Daniel Povey 2022-06-10 11:20:47 +08:00
parent 950cd4a3e8
commit c92d9d72aa
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py
View File

@ -901,6 +901,7 @@ class Decorrelate(torch.nn.Module):
if random.random() > apply_prob: if random.random() > apply_prob:
return x return x
with torch.cuda.amp.autocast(enabled=False): with torch.cuda.amp.autocast(enabled=False):
x = x.to(torch.float32)
ans = DecorrelateFunction.apply(x, self.cov.clone(), ans = DecorrelateFunction.apply(x, self.cov.clone(),
self.scale, self.eps, self.beta, self.scale, self.eps, self.beta,
self.channel_dim) # == x. self.channel_dim) # == x.