zhu-gu-an
2024-01-13 49e8e9d8fc1209c347aa2c2c65c6eb067b9f79d4
funasr/models/frontend/wav_frontend.py
@@ -145,9 +145,12 @@
            feats_lens.append(feat_length)
        feats_lens = torch.as_tensor(feats_lens)
        feats_pad = pad_sequence(feats,
                                 batch_first=True,
                                 padding_value=0.0)
        if batch_size == 1:
            feats_pad = feats[0][None, :, :]
        else:
            feats_pad = pad_sequence(feats,
                                     batch_first=True,
                                     padding_value=0.0)
        return feats_pad, feats_lens
    def forward_fbank(