This commit is contained in:
语帆 2024-03-01 11:30:49 +08:00
parent 583f918e0e
commit de21c10a2e

View File

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