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

Reduce onnxruntime log output. (#1306)

Browse Source 
Change the logging level from WARNING to ERROR.
This commit is contained in:
Fangjun Kuang
2024-08-30 12:50:34 +08:00
committed by GitHub
parent 6b8877f185
commit 3687c9f60a
17 changed files with 32 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sherpa-onnx/csrc/offline-ctc-model.cc
View File

@@ -35,7 +35,7 @@ namespace sherpa_onnx {
static ModelType GetModelType(char *model_data, size_t model_data_length, static ModelType GetModelType(char *model_data, size_t model_data_length,
bool debug) { bool debug) {
Ort::Env env(ORT_LOGGING_LEVEL_WARNING); Ort::Env env(ORT_LOGGING_LEVEL_ERROR);
Ort::SessionOptions sess_opts; Ort::SessionOptions sess_opts;
sess_opts.SetIntraOpNumThreads(1); sess_opts.SetIntraOpNumThreads(1);
sess_opts.SetInterOpNumThreads(1); sess_opts.SetInterOpNumThreads(1);

4
sherpa-onnx/csrc/offline-transducer-model.cc
View File

@@ -19,7 +19,7 @@ class OfflineTransducerModel::Impl {
public: public:
explicit Impl(const OfflineModelConfig &config) explicit Impl(const OfflineModelConfig &config)
: config_(config), : config_(config),
env_(ORT_LOGGING_LEVEL_WARNING), env_(ORT_LOGGING_LEVEL_ERROR),
sess_opts_(GetSessionOptions(config)), sess_opts_(GetSessionOptions(config)),
allocator_{} { allocator_{} {
{ {
@@ -41,7 +41,7 @@ class OfflineTransducerModel::Impl {
#if __ANDROID_API__ >= 9 #if __ANDROID_API__ >= 9
Impl(AAssetManager *mgr, const OfflineModelConfig &config) Impl(AAssetManager *mgr, const OfflineModelConfig &config)
: config_(config), : config_(config),
env_(ORT_LOGGING_LEVEL_WARNING), env_(ORT_LOGGING_LEVEL_ERROR),
sess_opts_(GetSessionOptions(config)), sess_opts_(GetSessionOptions(config)),
allocator_{} { allocator_{} {
{ {

4
sherpa-onnx/csrc/offline-transducer-nemo-model.cc
View File

@@ -21,7 +21,7 @@ class OfflineTransducerNeMoModel::Impl {
public: public:
explicit Impl(const OfflineModelConfig &config) explicit Impl(const OfflineModelConfig &config)
: config_(config), : config_(config),
env_(ORT_LOGGING_LEVEL_WARNING), env_(ORT_LOGGING_LEVEL_ERROR),
sess_opts_(GetSessionOptions(config)), sess_opts_(GetSessionOptions(config)),
allocator_{} { allocator_{} {
{ {
@@ -43,7 +43,7 @@ class OfflineTransducerNeMoModel::Impl {
#if __ANDROID_API__ >= 9 #if __ANDROID_API__ >= 9
Impl(AAssetManager *mgr, const OfflineModelConfig &config) Impl(AAssetManager *mgr, const OfflineModelConfig &config)
: config_(config), : config_(config),
env_(ORT_LOGGING_LEVEL_WARNING), env_(ORT_LOGGING_LEVEL_ERROR),
sess_opts_(GetSessionOptions(config)), sess_opts_(GetSessionOptions(config)),
allocator_{} { allocator_{} {
{ {

4
sherpa-onnx/csrc/offline-tts-vits-model.cc
View File

@@ -19,7 +19,7 @@ class OfflineTtsVitsModel::Impl {
public: public:
explicit Impl(const OfflineTtsModelConfig &config) explicit Impl(const OfflineTtsModelConfig &config)
: config_(config), : config_(config),
env_(ORT_LOGGING_LEVEL_WARNING), env_(ORT_LOGGING_LEVEL_ERROR),
sess_opts_(GetSessionOptions(config)), sess_opts_(GetSessionOptions(config)),
allocator_{} { allocator_{} {
auto buf = ReadFile(config.vits.model); auto buf = ReadFile(config.vits.model);
@@ -29,7 +29,7 @@ class OfflineTtsVitsModel::Impl {
#if __ANDROID_API__ >= 9 #if __ANDROID_API__ >= 9
Impl(AAssetManager *mgr, const OfflineTtsModelConfig &config) Impl(AAssetManager *mgr, const OfflineTtsModelConfig &config)
: config_(config), : config_(config),
env_(ORT_LOGGING_LEVEL_WARNING), env_(ORT_LOGGING_LEVEL_ERROR),
sess_opts_(GetSessionOptions(config)), sess_opts_(GetSessionOptions(config)),
allocator_{} { allocator_{} {
auto buf = ReadFile(mgr, config.vits.model); auto buf = ReadFile(mgr, config.vits.model);

4
sherpa-onnx/csrc/online-conformer-transducer-model.cc
View File

@@ -30,7 +30,7 @@ namespace sherpa_onnx {
OnlineConformerTransducerModel::OnlineConformerTransducerModel( OnlineConformerTransducerModel::OnlineConformerTransducerModel(
const OnlineModelConfig &config) const OnlineModelConfig &config)
: env_(ORT_LOGGING_LEVEL_WARNING), : env_(ORT_LOGGING_LEVEL_ERROR),
config_(config), config_(config),
sess_opts_(GetSessionOptions(config)), sess_opts_(GetSessionOptions(config)),
allocator_{} { allocator_{} {
@@ -53,7 +53,7 @@ OnlineConformerTransducerModel::OnlineConformerTransducerModel(
#if __ANDROID_API__ >= 9 #if __ANDROID_API__ >= 9
OnlineConformerTransducerModel::OnlineConformerTransducerModel( OnlineConformerTransducerModel::OnlineConformerTransducerModel(
AAssetManager *mgr, const OnlineModelConfig &config) AAssetManager *mgr, const OnlineModelConfig &config)
: env_(ORT_LOGGING_LEVEL_WARNING), : env_(ORT_LOGGING_LEVEL_ERROR),
config_(config), config_(config),
sess_opts_(GetSessionOptions(config)), sess_opts_(GetSessionOptions(config)),
allocator_{} { allocator_{} {

4
sherpa-onnx/csrc/online-lstm-transducer-model.cc
View File

@@ -28,7 +28,7 @@ namespace sherpa_onnx {
OnlineLstmTransducerModel::OnlineLstmTransducerModel( OnlineLstmTransducerModel::OnlineLstmTransducerModel(
const OnlineModelConfig &config) const OnlineModelConfig &config)
: env_(ORT_LOGGING_LEVEL_WARNING), : env_(ORT_LOGGING_LEVEL_ERROR),
config_(config), config_(config),
sess_opts_(GetSessionOptions(config)), sess_opts_(GetSessionOptions(config)),
allocator_{} { allocator_{} {
@@ -51,7 +51,7 @@ OnlineLstmTransducerModel::OnlineLstmTransducerModel(
#if __ANDROID_API__ >= 9 #if __ANDROID_API__ >= 9
OnlineLstmTransducerModel::OnlineLstmTransducerModel( OnlineLstmTransducerModel::OnlineLstmTransducerModel(
AAssetManager *mgr, const OnlineModelConfig &config) AAssetManager *mgr, const OnlineModelConfig &config)
: env_(ORT_LOGGING_LEVEL_WARNING), : env_(ORT_LOGGING_LEVEL_ERROR),
config_(config), config_(config),
sess_opts_(GetSessionOptions(config)), sess_opts_(GetSessionOptions(config)),
allocator_{} { allocator_{} {

2
sherpa-onnx/csrc/online-nemo-ctc-model.cc
View File

@@ -39,7 +39,7 @@ class OnlineNeMoCtcModel::Impl {
#if __ANDROID_API__ >= 9 #if __ANDROID_API__ >= 9
Impl(AAssetManager *mgr, const OnlineModelConfig &config) Impl(AAssetManager *mgr, const OnlineModelConfig &config)
: config_(config), : config_(config),
env_(ORT_LOGGING_LEVEL_WARNING), env_(ORT_LOGGING_LEVEL_ERROR),
sess_opts_(GetSessionOptions(config)), sess_opts_(GetSessionOptions(config)),
allocator_{} { allocator_{} {
{ {

2
sherpa-onnx/csrc/online-paraformer-model.cc
View File

@@ -41,7 +41,7 @@ class OnlineParaformerModel::Impl {
#if __ANDROID_API__ >= 9 #if __ANDROID_API__ >= 9
Impl(AAssetManager *mgr, const OnlineModelConfig &config) Impl(AAssetManager *mgr, const OnlineModelConfig &config)
: config_(config), : config_(config),
env_(ORT_LOGGING_LEVEL_WARNING), env_(ORT_LOGGING_LEVEL_ERROR),
sess_opts_(GetSessionOptions(config)), sess_opts_(GetSessionOptions(config)),
allocator_{} { allocator_{} {
{ {

4
sherpa-onnx/csrc/online-recognizer-impl.cc
View File

@@ -28,7 +28,7 @@ namespace sherpa_onnx {
std::unique_ptr<OnlineRecognizerImpl> OnlineRecognizerImpl::Create( std::unique_ptr<OnlineRecognizerImpl> OnlineRecognizerImpl::Create(
const OnlineRecognizerConfig &config) { const OnlineRecognizerConfig &config) {
if (!config.model_config.transducer.encoder.empty()) { if (!config.model_config.transducer.encoder.empty()) {
Ort::Env env(ORT_LOGGING_LEVEL_WARNING); Ort::Env env(ORT_LOGGING_LEVEL_ERROR);
auto decoder_model = ReadFile(config.model_config.transducer.decoder); auto decoder_model = ReadFile(config.model_config.transducer.decoder);
auto sess = std::make_unique<Ort::Session>( auto sess = std::make_unique<Ort::Session>(
@@ -61,7 +61,7 @@ std::unique_ptr<OnlineRecognizerImpl> OnlineRecognizerImpl::Create(
std::unique_ptr<OnlineRecognizerImpl> OnlineRecognizerImpl::Create( std::unique_ptr<OnlineRecognizerImpl> OnlineRecognizerImpl::Create(
AAssetManager *mgr, const OnlineRecognizerConfig &config) { AAssetManager *mgr, const OnlineRecognizerConfig &config) {
if (!config.model_config.transducer.encoder.empty()) { if (!config.model_config.transducer.encoder.empty()) {
Ort::Env env(ORT_LOGGING_LEVEL_WARNING); Ort::Env env(ORT_LOGGING_LEVEL_ERROR);
auto decoder_model = ReadFile(mgr, config.model_config.transducer.decoder); auto decoder_model = ReadFile(mgr, config.model_config.transducer.decoder);
auto sess = std::make_unique<Ort::Session>( auto sess = std::make_unique<Ort::Session>(

2
sherpa-onnx/csrc/online-transducer-model.cc
View File

@@ -37,7 +37,7 @@ namespace sherpa_onnx {
static ModelType GetModelType(char *model_data, size_t model_data_length, static ModelType GetModelType(char *model_data, size_t model_data_length,
bool debug) { bool debug) {
Ort::Env env(ORT_LOGGING_LEVEL_WARNING); Ort::Env env(ORT_LOGGING_LEVEL_ERROR);
Ort::SessionOptions sess_opts; Ort::SessionOptions sess_opts;
sess_opts.SetIntraOpNumThreads(1); sess_opts.SetIntraOpNumThreads(1);
sess_opts.SetInterOpNumThreads(1); sess_opts.SetInterOpNumThreads(1);

4
sherpa-onnx/csrc/online-transducer-nemo-model.cc
View File

@@ -35,7 +35,7 @@ class OnlineTransducerNeMoModel::Impl {
public: public:
explicit Impl(const OnlineModelConfig &config) explicit Impl(const OnlineModelConfig &config)
: config_(config), : config_(config),
env_(ORT_LOGGING_LEVEL_WARNING), env_(ORT_LOGGING_LEVEL_ERROR),
sess_opts_(GetSessionOptions(config)), sess_opts_(GetSessionOptions(config)),
allocator_{} { allocator_{} {
{ {
@@ -57,7 +57,7 @@ class OnlineTransducerNeMoModel::Impl {
#if __ANDROID_API__ >= 9 #if __ANDROID_API__ >= 9
Impl(AAssetManager *mgr, const OnlineModelConfig &config) Impl(AAssetManager *mgr, const OnlineModelConfig &config)
: config_(config), : config_(config),
env_(ORT_LOGGING_LEVEL_WARNING), env_(ORT_LOGGING_LEVEL_ERROR),
sess_opts_(GetSessionOptions(config)), sess_opts_(GetSessionOptions(config)),
allocator_{} { allocator_{} {
{ {

2
sherpa-onnx/csrc/online-wenet-ctc-model.cc
View File

@@ -36,7 +36,7 @@ class OnlineWenetCtcModel::Impl {
#if __ANDROID_API__ >= 9 #if __ANDROID_API__ >= 9
Impl(AAssetManager *mgr, const OnlineModelConfig &config) Impl(AAssetManager *mgr, const OnlineModelConfig &config)
: config_(config), : config_(config),
env_(ORT_LOGGING_LEVEL_WARNING), env_(ORT_LOGGING_LEVEL_ERROR),
sess_opts_(GetSessionOptions(config)), sess_opts_(GetSessionOptions(config)),
allocator_{} { allocator_{} {
{ {

4
sherpa-onnx/csrc/online-zipformer-transducer-model.cc
View File

@@ -30,7 +30,7 @@ namespace sherpa_onnx {
OnlineZipformerTransducerModel::OnlineZipformerTransducerModel( OnlineZipformerTransducerModel::OnlineZipformerTransducerModel(
const OnlineModelConfig &config) const OnlineModelConfig &config)
: env_(ORT_LOGGING_LEVEL_WARNING), : env_(ORT_LOGGING_LEVEL_ERROR),
config_(config), config_(config),
sess_opts_(GetSessionOptions(config)), sess_opts_(GetSessionOptions(config)),
allocator_{} { allocator_{} {
@@ -53,7 +53,7 @@ OnlineZipformerTransducerModel::OnlineZipformerTransducerModel(
#if __ANDROID_API__ >= 9 #if __ANDROID_API__ >= 9
OnlineZipformerTransducerModel::OnlineZipformerTransducerModel( OnlineZipformerTransducerModel::OnlineZipformerTransducerModel(
AAssetManager *mgr, const OnlineModelConfig &config) AAssetManager *mgr, const OnlineModelConfig &config)
: env_(ORT_LOGGING_LEVEL_WARNING), : env_(ORT_LOGGING_LEVEL_ERROR),
config_(config), config_(config),
sess_opts_(GetSessionOptions(config)), sess_opts_(GetSessionOptions(config)),
allocator_{} { allocator_{} {

2
sherpa-onnx/csrc/online-zipformer2-ctc-model.cc
View File

@@ -40,7 +40,7 @@ class OnlineZipformer2CtcModel::Impl {
#if __ANDROID_API__ >= 9 #if __ANDROID_API__ >= 9
Impl(AAssetManager *mgr, const OnlineModelConfig &config) Impl(AAssetManager *mgr, const OnlineModelConfig &config)
: config_(config), : config_(config),
env_(ORT_LOGGING_LEVEL_WARNING), env_(ORT_LOGGING_LEVEL_ERROR),
sess_opts_(GetSessionOptions(config)), sess_opts_(GetSessionOptions(config)),
allocator_{} { allocator_{} {
{ {

16
sherpa-onnx/csrc/online-zipformer2-transducer-model.cc
View File

@@ -32,7 +32,7 @@ namespace sherpa_onnx {
OnlineZipformer2TransducerModel::OnlineZipformer2TransducerModel( OnlineZipformer2TransducerModel::OnlineZipformer2TransducerModel(
const OnlineModelConfig &config) const OnlineModelConfig &config)
: env_(ORT_LOGGING_LEVEL_WARNING), : env_(ORT_LOGGING_LEVEL_ERROR),
encoder_sess_opts_(GetSessionOptions(config)), encoder_sess_opts_(GetSessionOptions(config)),
decoder_sess_opts_(GetSessionOptions(config, "decoder")), decoder_sess_opts_(GetSessionOptions(config, "decoder")),
joiner_sess_opts_(GetSessionOptions(config, "joiner")), joiner_sess_opts_(GetSessionOptions(config, "joiner")),
@@ -57,7 +57,7 @@ OnlineZipformer2TransducerModel::OnlineZipformer2TransducerModel(
#if __ANDROID_API__ >= 9 #if __ANDROID_API__ >= 9
OnlineZipformer2TransducerModel::OnlineZipformer2TransducerModel( OnlineZipformer2TransducerModel::OnlineZipformer2TransducerModel(
AAssetManager *mgr, const OnlineModelConfig &config) AAssetManager *mgr, const OnlineModelConfig &config)
: env_(ORT_LOGGING_LEVEL_WARNING), : env_(ORT_LOGGING_LEVEL_ERROR),
config_(config), config_(config),
encoder_sess_opts_(GetSessionOptions(config)), encoder_sess_opts_(GetSessionOptions(config)),
decoder_sess_opts_(GetSessionOptions(config)), decoder_sess_opts_(GetSessionOptions(config)),
@@ -82,8 +82,8 @@ OnlineZipformer2TransducerModel::OnlineZipformer2TransducerModel(
void OnlineZipformer2TransducerModel::InitEncoder(void *model_data, void OnlineZipformer2TransducerModel::InitEncoder(void *model_data,
size_t model_data_length) { size_t model_data_length) {
encoder_sess_ = std::make_unique<Ort::Session>(env_, model_data, encoder_sess_ = std::make_unique<Ort::Session>(
model_data_length, encoder_sess_opts_); env_, model_data, model_data_length, encoder_sess_opts_);
GetInputNames(encoder_sess_.get(), &encoder_input_names_, GetInputNames(encoder_sess_.get(), &encoder_input_names_,
&encoder_input_names_ptr_); &encoder_input_names_ptr_);
@@ -135,8 +135,8 @@ void OnlineZipformer2TransducerModel::InitEncoder(void *model_data,
void OnlineZipformer2TransducerModel::InitDecoder(void *model_data, void OnlineZipformer2TransducerModel::InitDecoder(void *model_data,
size_t model_data_length) { size_t model_data_length) {
decoder_sess_ = std::make_unique<Ort::Session>(env_, model_data, decoder_sess_ = std::make_unique<Ort::Session>(
model_data_length, decoder_sess_opts_); env_, model_data, model_data_length, decoder_sess_opts_);
GetInputNames(decoder_sess_.get(), &decoder_input_names_, GetInputNames(decoder_sess_.get(), &decoder_input_names_,
&decoder_input_names_ptr_); &decoder_input_names_ptr_);
@@ -160,8 +160,8 @@ void OnlineZipformer2TransducerModel::InitDecoder(void *model_data,
void OnlineZipformer2TransducerModel::InitJoiner(void *model_data, void OnlineZipformer2TransducerModel::InitJoiner(void *model_data,
size_t model_data_length) { size_t model_data_length) {
joiner_sess_ = std::make_unique<Ort::Session>(env_, model_data, joiner_sess_ = std::make_unique<Ort::Session>(
model_data_length, joiner_sess_opts_); env_, model_data, model_data_length, joiner_sess_opts_);
GetInputNames(joiner_sess_.get(), &joiner_input_names_, GetInputNames(joiner_sess_.get(), &joiner_input_names_,
&joiner_input_names_ptr_); &joiner_input_names_ptr_);

2
sherpa-onnx/csrc/speaker-embedding-extractor-impl.cc
View File

@@ -23,7 +23,7 @@ enum class ModelType : std::uint8_t {
static ModelType GetModelType(char *model_data, size_t model_data_length, static ModelType GetModelType(char *model_data, size_t model_data_length,
bool debug) { bool debug) {
Ort::Env env(ORT_LOGGING_LEVEL_WARNING); Ort::Env env(ORT_LOGGING_LEVEL_ERROR);
Ort::SessionOptions sess_opts; Ort::SessionOptions sess_opts;
sess_opts.SetIntraOpNumThreads(1); sess_opts.SetIntraOpNumThreads(1);
sess_opts.SetInterOpNumThreads(1); sess_opts.SetInterOpNumThreads(1);

2
sherpa-onnx/csrc/spoken-language-identification-impl.cc
View File

@@ -27,7 +27,7 @@ enum class ModelType : std::uint8_t {
static ModelType GetModelType(char *model_data, size_t model_data_length, static ModelType GetModelType(char *model_data, size_t model_data_length,
bool debug) { bool debug) {
Ort::Env env(ORT_LOGGING_LEVEL_WARNING); Ort::Env env(ORT_LOGGING_LEVEL_ERROR);
Ort::SessionOptions sess_opts; Ort::SessionOptions sess_opts;
auto sess = std::make_unique<Ort::Session>(env, model_data, model_data_length, auto sess = std::make_unique<Ort::Session>(env, model_data, model_data_length,