FUNASR训练
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
Yuekai Zhang
2024-11-12 ef1d7b3f12535d73bf9046085d14d192c54cf7da 
 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);