FUNASR训练
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
inisis
2024-12-19 1368a9bca48473d793f0c1d52df0a4739c747d16 
 runtime/onnxruntime/src/paraformer.h


@@ -42,11 +42,12 @@


    public:


        Paraformer();


        ~Paraformer();


        void InitAsr(const std::string &am_model, const std::string &am_cmvn, const std::string &am_config, int thread_num);


        void InitAsr(const std::string &am_model, const std::string &am_cmvn, const std::string &am_config, const std::string &token_file, int thread_num);


        // online


        void InitAsr(const std::string &en_model, const std::string &de_model, const std::string &am_cmvn, const std::string &am_config, int thread_num);


        void InitAsr(const std::string &en_model, const std::string &de_model, const std::string &am_cmvn, const std::string &am_config, const std::string &token_file, int thread_num);


        // 2pass


        void InitAsr(const std::string &am_model, const std::string &en_model, const std::string &de_model, const std::string &am_cmvn, const std::string &am_config, int thread_num);


        void InitAsr(const std::string &am_model, const std::string &en_model, const std::string &de_model, const std::string &am_cmvn, 


            const std::string &am_config, const std::string &token_file, const std::string &online_token_file, int thread_num);


        void InitHwCompiler(const std::string &hw_model, int thread_num);


        void InitSegDict(const std::string &seg_dict_model);


        std::vector<std::vector<float>> CompileHotwordEmbedding(std::string &hotwords);