FUNASR训练
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
Lizerui9926
2023-04-26 b78d47f1efb3d0662fce1b8d45a9eb11b3caef02 
 funasr/runtime/onnxruntime/src/ct-transformer.h


@@ -1,3 +1,8 @@


/**


 * Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights Reserved.


 * MIT License  (https://opensource.org/licenses/MIT)


*/




#pragma once 




class CTTransformer {


@@ -19,7 +24,8 @@


    Ort::SessionOptions session_options;


public:




   CTTransformer(const char* sz_model_dir, int thread_num);


   CTTransformer();


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


   ~CTTransformer();


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


   string AddPunc(const char* sz_input);