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

Bug fix RE GPU device

Browse Source
This commit is contained in:
Daniel Povey 2022-06-06 15:40:20 +08:00
parent 71e927411a
commit 6fdb356315
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

@ -657,7 +657,7 @@ class FixedProjDrop(torch.nn.Module):
rand_mat = torch.randn(num_channels, num_channels)
U, _, _ = rand_mat.svd()
self.U = U # a random orthogonal square matrix. will be a buffer.
self.register_buffer('U', U) # a random orthogonal square matrix. will be a buffer.
def forward(self, x: Tensor) -> Tensor: