This commit is contained in:
语帆 2024-03-01 11:18:29 +08:00
parent 6d7b945710
commit 583f918e0e

View File

@ -428,7 +428,10 @@ class LCBNet(nn.Module):
meta_data["load_data"] = f"{time2 - time1:0.3f}"
pdb.set_trace()
audio_sample_list = sample_list[0]
ocr_sample_list = sample_list[1]
if len(sample_list) >1:
ocr_sample_list = sample_list[1]
else:
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()