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

Bug fix

Browse Source
This commit is contained in:
Daniel Povey 2022-08-02 14:31:37 +08:00
parent e9f4ada1c0
commit e44ab25e99
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_stateless7/optim.py
View File

@ -677,7 +677,7 @@ param_rms_smooth1: Smoothing proportion for parameter matrix, if assumed rank of
# value) to be 1, and since the singular values (==eigenvalues) of P
# are the squares of the singular values of C, we want them to have
# mean of 1.
P /= _mean(_diag(P), exclude_dims=[0]).unsqueeze(-1)
P /= _mean(_diag(P), exclude_dims=[0], keepdim=True).unsqueeze(-1)
# C will satisfy: P == torch.matmul(C, C.transpose(2, 3))
# C is of shape (batch_size, num_blocks, block_size, block_size).