Shi Xian
2024-12-05 0efc87352ce7d3903dbdedbfa5d01ca5e1cb19e7
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