Przeglądaj źródła

Update e2e_asr_transducer.py

loss_ctc bug fix.
aky15 2 lat temu
rodzic
commit
ecccb37896
1 zmienionych plików z 2 dodań i 2 usunięć
  1. 2 2
      funasr/models/e2e_asr_transducer.py

+ 2 - 2
funasr/models/e2e_asr_transducer.py

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