EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Adding warm up for Zipformer2 (#766)

Browse Source 
Signed-off-by: manickavela1998@gmail.com <manickavela1998@gmail.com>
This commit is contained in:
Manix
2024-04-16 06:46:55 +05:30
committed by GitHub
parent 5981adf454
commit fb4aee83ac
9 changed files with 99 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
sherpa-onnx/csrc/online-model-config.cc
View File

@@ -21,6 +21,10 @@ void OnlineModelConfig::Register(ParseOptions *po) {
po->Register("num-threads", &num_threads,
"Number of threads to run the neural network");
po->Register("warm-up", &warm_up,
"Number of warm-up to run the onnxruntime"
"Valid vales are: zipformer2");
po->Register("debug", &debug,
"true to print model information while loading it.");
@@ -70,6 +74,7 @@ std::string OnlineModelConfig::ToString() const {
os << "zipformer2_ctc=" << zipformer2_ctc.ToString() << ", ";
os << "tokens=\"" << tokens << "\", ";
os << "num_threads=" << num_threads << ", ";
os << "warm_up=" << warm_up << ", ";
os << "debug=" << (debug ? "True" : "False") << ", ";
os << "provider=\"" << provider << "\", ";
os << "model_type=\"" << model_type << "\")";