游雁
2023-04-24 4d72ada6a23f68baa3ee59ea994cf0ceaab9da64
funasr/modules/e2e_asr_common.py
@@ -18,8 +18,7 @@
import torch
from funasr.modules.beam_search.beam_search_transducer import BeamSearchTransducer
from funasr.models.rnnt_decoder.abs_decoder import AbsDecoder
from funasr.models.joint_network import JointNetwork
from funasr.models.joint_net.joint_network import JointNetwork
def end_detect(ended_hyps, i, M=3, D_end=np.log(1 * np.exp(-10))):
    """End detection.
@@ -268,7 +267,7 @@
    def __init__(
        self,
        decoder: AbsDecoder,
        decoder,
        joint_network: JointNetwork,
        token_list: List[int],
        sym_space: str,