mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-19 05:54:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

CTC loss return tensor

Browse Source
This commit is contained in:
Yifan Yang 2023-06-16 14:52:39 +08:00 committed by GitHub
parent 0a465794a8
commit 79b877f037
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/librispeech/ASR/zipformer/model.py
View File

@ -353,6 +353,6 @@ class AsrModel(nn.Module):
target_lengths=y_lens,
)
else:
ctc_loss = 0
ctc_loss = torch.zeros(0)
return simple_loss, pruned_loss, ctc_loss