o il@s@ddlmZddlmZmZmZeddGdddeZdS))tables)ERes2NetBasicBlockERes2NetBasicBlockERes2Net_diff_AFFlid_predictor_classes LidPredictorcs2eZdZeegddddddffdd ZZS) r)r PTSTPFc s$tt|j||||||||ddS)N)block block_fuse num_blocks m_channelsfeat_dimembedding_size pooling_func two_emb_layer)superr__init__) selfrrrrrrrr __class__[/home/ubuntu/.local/lib/python3.10/site-packages/funasr/models/whisper_lid/lid_predictor.pyr s  zLidPredictor.__init__)__name__ __module__ __qualname__rrr __classcell__rrrrr sN)funasr.registerr)funasr.models.whisper_lid.eres2net.ResNetrrrregisterrrrrrs