语帆 2 년 전
부모
커밋
de21c10a2e
1개의 변경된 파일1개의 추가작업 그리고 2개의 파일을 삭제
  1. 1 2
      funasr/models/lcbnet/model.py

+ 1 - 2
funasr/models/lcbnet/model.py

@@ -426,12 +426,11 @@ class LCBNet(nn.Module):
                                                             tokenizer=tokenizer)
             time2 = time.perf_counter()
             meta_data["load_data"] = f"{time2 - time1:0.3f}"
-            pdb.set_trace()
             audio_sample_list = sample_list[0]
             if len(sample_list) >1:
                 ocr_sample_list = sample_list[1]
             else:
-                ocr_sample_list = [294, 0]
+                ocr_sample_list = [[294, 0]]
             speech, speech_lengths = extract_fbank(audio_sample_list, data_type=kwargs.get("data_type", "sound"),
                                                    frontend=frontend)
             time3 = time.perf_counter()