FUNASR训练
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
Truco
2024-10-28 1a45b647a858b72c711c64bfe6b2e5333df6fd86 
 runtime/onnxruntime/src/paraformer.h


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


    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, 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);