FUNASR训练
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
游雁
2024-05-15 67733a2a92f54d9777a0b18ac4f1488b3bc41efc 
 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);