FUNASR训练
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
游雁
2024-05-20 73f3f2f91b8549371d8a62ca41355a301d6fcc50 
 runtime/onnxruntime/src/ct-transformer.h


@@ -26,7 +26,7 @@


public:




   CTTransformer();


   void InitPunc(const std::string &punc_model, const std::string &punc_config, int thread_num);


   void InitPunc(const std::string &punc_model, const std::string &punc_config, const std::string &token_file, int thread_num);


   ~CTTransformer();


   vector<int>  Infer(vector<int32_t> input_data);


   string AddPunc(const char* sz_input, std::string language="zh-cn");