Procházet zdrojové kódy

fix uniasr training bug

仁迷 před 3 roky
rodič
revize
c5c5c55b90
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      funasr/models/e2e_uni_asr.py

+ 1 - 1
funasr/models/e2e_uni_asr.py

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