speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update e2e_asr_transducer.py

Browse Source 
loss_ctc bug fix.
This commit is contained in:
aky15 2023-08-11 10:54:17 +08:00 committed by GitHub
parent 4fd7f02b26
commit ecccb37896
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
funasr/models/e2e_asr_transducer.py
View File

@ -697,7 +697,7 @@ class UnifiedTransducerModel(FunASRModel):
loss_trans = loss_trans_utt + loss_trans_chunk loss_trans = loss_trans_utt + loss_trans_chunk
loss_ctc = loss_ctc + loss_ctc_chunk loss_ctc = loss_ctc + loss_ctc_chunk
loss_ctc = loss_att + loss_att_chunk loss_att = loss_att + loss_att_chunk
loss = ( loss = (
self.transducer_weight * loss_trans self.transducer_weight * loss_trans
@ -990,4 +990,4 @@ class UnifiedTransducerModel(FunASRModel):
ignore_label=self.ignore_id, ignore_label=self.ignore_id,
) )
return loss_att, acc_att return loss_att, acc_att