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

Update train.py (#1299)

Browse Source
This commit is contained in:
zr_jin 2023-10-11 10:06:00 +08:00 committed by GitHub
parent 9af144c26b
commit 0d09a44930
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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/aishell/ASR/pruned_transducer_stateless7/train.py
View File

@ -703,7 +703,7 @@ def compute_loss(
if batch_idx_train >= warm_step if batch_idx_train >= warm_step
else 0.1 + 0.9 * (batch_idx_train / warm_step) else 0.1 + 0.9 * (batch_idx_train / warm_step)
) )
loss = params.simple_loss_scale * simple_loss + pruned_loss_scale * pruned_loss loss = simple_loss_scale * simple_loss + pruned_loss_scale * pruned_loss
assert loss.requires_grad == is_training assert loss.requires_grad == is_training