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

Reduce initial pruned_loss scale from 0.01 to 0.0

Browse Source
This commit is contained in:
Daniel Povey 2022-03-22 12:30:48 +08:00
parent b7e84d5d77
commit b82a505dfc
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/train.py
View File

@ -496,7 +496,7 @@ def compute_loss(
warmup_mode=warmup_mode, warmup_mode=warmup_mode,
) )
loss = (params.simple_loss_scale * simple_loss + loss = (params.simple_loss_scale * simple_loss +
(pruned_loss * 0.01 if warmup_mode else pruned_loss)) (pruned_loss * 0.0 if warmup_mode else pruned_loss))
assert loss.requires_grad == is_training assert loss.requires_grad == is_training