莫拉古
2024-11-29 ae49b2a8e1bc676e6014d8a12ebeec947b655e3e
funasr/frontends/utils/log_mel.py
@@ -73,11 +73,7 @@
        # Zero padding
        if ilens is not None:
            logmel_feat = logmel_feat.masked_fill(
                make_pad_mask(ilens, logmel_feat, 1), 0.0
            )
            logmel_feat = logmel_feat.masked_fill(make_pad_mask(ilens, logmel_feat, 1), 0.0)
        else:
            ilens = feat.new_full(
                [feat.size(0)], fill_value=feat.size(1), dtype=torch.long
            )
            ilens = feat.new_full([feat.size(0)], fill_value=feat.size(1), dtype=torch.long)
        return logmel_feat, ilens