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

Fix bug that relates to modifying U in place

Browse Source
This commit is contained in:
Daniel Povey 2022-06-06 17:43:15 +08:00
parent 31848dcd11
commit 4352a16f57
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -668,11 +668,11 @@ class RandProjDrop(torch.nn.Module):
else: else:
self._randomize_U() self._randomize_U()
x = x.transpose(self.channel_dim, -1) # (..., num_channels) x = x.transpose(self.channel_dim, -1) # (..., num_channels)
U = self.U.clone()
x = torch.matmul(x, self.U) x = torch.matmul(x, U)
x = torch.nn.functional.dropout(x, self.dropout_rate, x = torch.nn.functional.dropout(x, self.dropout_rate,
training=True) training=True)
x = torch.matmul(x, self.U.t()) x = torch.matmul(x, U.t())
x = x.transpose(self.channel_dim, -1) # (..., num_channels) x = x.transpose(self.channel_dim, -1) # (..., num_channels)
return x return x