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

Make it work for half

Browse Source
This commit is contained in:
Daniel Povey 2022-06-05 23:25:51 +08:00
parent e535887abb
commit d76aedb790
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/pruned_transducer_stateless2/scaling.py
View File

@ -667,7 +667,7 @@ class ProjDrop(torch.nn.Module):
device=x.device, dtype=x.dtype)
rr = torch.matmul(r, r.t()) # num_dropped by num_dropped
rr += 0.01 # to 100% ensure it is invertible
rr_inv = rr.cholesky().cholesky_inverse()
rr_inv = rr.cholesky().to(torch.float32).cholesky_inverse().to(x.dtype)
# OK, so r rr_inv r.t() will have eigenvalues of 1.
xr = torch.matmul(x, r.t()) # (..., num_dropped)
rr_inv_r = torch.matmul(rr_inv, r) # (num_dropped, num_channels)