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

test

Browse Source
This commit is contained in:
语帆 2024-02-22 17:31:57 +08:00
parent b349739f5d
commit e615585fd3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
funasr/models/lcbnet/model.py
View File

@ -89,7 +89,7 @@ class LCBNet(nn.Module):
text_encoder = text_encoder_class(input_size=vocab_size, **text_encoder_conf)
fusion_encoder_class = tables.encoder_classes.get(fusion_encoder)
fusion_encoder = fusion_encoder_class(**fusion_encoder_conf)
bias_predictor_class = tables.encoder_classes.get_class(bias_predictor)
bias_predictor_class = tables.encoder_classes.get(bias_predictor)
bias_predictor = bias_predictor_class(bias_predictor_conf)
if decoder is not None:
@ -414,7 +414,7 @@ class LCBNet(nn.Module):
self.init_beam_search(**kwargs)
self.nbest = kwargs.get("nbest", 1)
pdb.set_trace()
meta_data = {}
if isinstance(data_in, torch.Tensor) and kwargs.get("data_type", "sound") == "fbank": # fbank
speech, speech_lengths = data_in, data_lengths