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

Fix bug

Browse Source
This commit is contained in:
Daniel Povey 2022-03-25 20:35:11 +08:00
parent 4b650e9f01
commit d2ed3dfc90
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/pruned_transducer_stateless2/train.py
View File

@ -506,8 +506,8 @@ def compute_loss(
# overwhelming the simple_loss and causing it to diverge,
# in case it had not fully learned the alignment yet.
pruned_loss_scale = (0.0 if warmup < 1.0 else
(0.1 if warmup > 1.0 and warmup < 2.0) else
1.0)
(0.1 if warmup > 1.0 and warmup < 2.0 else
1.0))
loss = (params.simple_loss_scale * simple_loss +
pruned_loss_scale * pruned_loss)