mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update model.py

Browse Source
This commit is contained in:
Yifan Yang 2023-06-16 14:56:40 +08:00 committed by GitHub
parent 79b877f037
commit fc6c64e029
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
egs/librispeech/ASR/zipformer/model.py
View File

@ -340,8 +340,8 @@ class AsrModel(nn.Module):
lm_scale=lm_scale, lm_scale=lm_scale,
) )
else: else:
simple_loss = 0 simple_loss = torch.empty(0)
pruned_loss = 0 pruned_loss = torch.empty(0)
if self.use_ctc: if self.use_ctc:
# Compute CTC loss # Compute CTC loss
@ -353,6 +353,6 @@ class AsrModel(nn.Module):
target_lengths=y_lens, target_lengths=y_lens,
) )
else: else:
ctc_loss = torch.zeros(0) ctc_loss = torch.empty(0)
return simple_loss, pruned_loss, ctc_loss return simple_loss, pruned_loss, ctc_loss