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

Merge pull request #82 from alibaba-damo-academy/dev_lhn

Browse Source 
fix uniasr training bug
This commit is contained in:
hnluo 2023-02-09 15:17:19 +08:00 committed by GitHub
commit 1bae17a7a9
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
funasr/models/e2e_uni_asr.py
View File

@ -206,7 +206,7 @@ class UniASR(AbsESPnetModel):
with torch.no_grad():
speech_raw, encoder_out, encoder_out_lens = self.encode(speech, speech_lengths, ind=ind)
else:
speech_raw, encoder_out_lens = self.encode(speech, speech_lengths, ind=ind)
speech_raw, encoder_out, encoder_out_lens = self.encode(speech, speech_lengths, ind=ind)
intermediate_outs = None
if isinstance(encoder_out, tuple):