lzr265946
2022-12-03 a9e857e45250b16af60d5fe3efcd06e685f6506a
funasr/models/predictor/cif.py
@@ -4,7 +4,7 @@
from funasr.modules.nets_utils import make_pad_mask
class CifPredictor(nn.Module):
    def __init__(self, idim, l_order, r_order, threshold=1.0, dropout=0.1, smooth_factor=1.0, noise_threshold=0):
    def __init__(self, idim, l_order, r_order, threshold=1.0, dropout=0.1, smooth_factor=1.0, noise_threshold=0, tail_threshold=0.45):
        super(CifPredictor, self).__init__()
        self.pad = nn.ConstantPad1d((l_order, r_order), 0)