mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-09 01:52:41 +00:00
Update egs/aishell/ASR/zipformer/train.py
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
This commit is contained in:
parent
b20c4b5f46
commit
a063afb349
@ -856,7 +856,10 @@ def compute_loss(
|
||||
supervision_segments=supervision_segments,
|
||||
time_warp_factor=params.spec_aug_time_warp_factor,
|
||||
)
|
||||
if params.use_ctc:
|
||||
simple_loss, pruned_loss, ctc_loss, _, cr_loss = losses[:5]
|
||||
else:
|
||||
simple_loss, pruned_loss = losses[:2]
|
||||
|
||||
loss = 0.0
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user