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

Update egs/aishell/ASR/zipformer/train.py

Browse Source 
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
This commit is contained in:
Mistmoon 2025-07-07 10:15:44 +08:00 committed by GitHub
parent b20c4b5f46
commit a063afb349
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
egs/aishell/ASR/zipformer/train.py
View File

@ -856,7 +856,10 @@ def compute_loss(
supervision_segments=supervision_segments, supervision_segments=supervision_segments,
time_warp_factor=params.spec_aug_time_warp_factor, time_warp_factor=params.spec_aug_time_warp_factor,
) )
simple_loss, pruned_loss, ctc_loss, _, cr_loss = losses[:5] if params.use_ctc:
simple_loss, pruned_loss, ctc_loss, _, cr_loss = losses[:5]
else:
simple_loss, pruned_loss = losses[:2]
loss = 0.0 loss = 0.0