From 11ae5744509286cf517758bf99ceda58c9ff2425 Mon Sep 17 00:00:00 2001
From: lyblsgo <lyblsgo@163.com>
Date: 星期四, 20 四月 2023 15:52:11 +0800
Subject: [PATCH] fix code style
---
funasr/runtime/onnxruntime/src/e2e_vad.h | 3 ---
funasr/runtime/onnxruntime/src/paraformer_onnx.cpp | 6 +++---
funasr/runtime/onnxruntime/src/FsmnVad.h | 3 ---
funasr/runtime/onnxruntime/src/FsmnVad.cc | 9 +--------
4 files changed, 4 insertions(+), 17 deletions(-)
diff --git a/funasr/runtime/onnxruntime/src/FsmnVad.cc b/funasr/runtime/onnxruntime/src/FsmnVad.cc
index 69c69ca..6de482e 100644
--- a/funasr/runtime/onnxruntime/src/FsmnVad.cc
+++ b/funasr/runtime/onnxruntime/src/FsmnVad.cc
@@ -1,6 +1,4 @@
-//
-// Created by root on 4/9/23.
-//
+
#include <fstream>
#include "FsmnVad.h"
#include "precomp.h"
@@ -9,8 +7,6 @@
void FsmnVad::init_vad(const std::string &vad_model, const std::string &vad_cmvn, int vad_sample_rate, int vad_silence_duration, int vad_max_len,
float vad_speech_noise_thres) {
-
-
session_options_.SetIntraOpNumThreads(1);
session_options_.SetGraphOptimizationLevel(ORT_ENABLE_ALL);
session_options_.DisableCpuMemArena();
@@ -33,7 +29,6 @@
}
-
void FsmnVad::read_model(const std::string &vad_model) {
try {
vad_session_ = std::make_shared<Ort::Session>(
@@ -45,8 +40,6 @@
//LOG(INFO) << "vad onnx:";
GetInputOutputInfo(vad_session_, &vad_in_names_, &vad_out_names_);
}
-
-
void FsmnVad::GetInputOutputInfo(
const std::shared_ptr<Ort::Session> &session,
diff --git a/funasr/runtime/onnxruntime/src/FsmnVad.h b/funasr/runtime/onnxruntime/src/FsmnVad.h
index 839f9ec..69a1b98 100644
--- a/funasr/runtime/onnxruntime/src/FsmnVad.h
+++ b/funasr/runtime/onnxruntime/src/FsmnVad.h
@@ -1,6 +1,3 @@
-//
-// Created by zyf on 4/9/23.
-//
#ifndef VAD_SERVER_FSMNVAD_H
#define VAD_SERVER_FSMNVAD_H
diff --git a/funasr/runtime/onnxruntime/src/e2e_vad.h b/funasr/runtime/onnxruntime/src/e2e_vad.h
index b1780a7..f0c4975 100644
--- a/funasr/runtime/onnxruntime/src/e2e_vad.h
+++ b/funasr/runtime/onnxruntime/src/e2e_vad.h
@@ -1,6 +1,3 @@
-//
-// Created by root on 3/31/23.
-//
#include <utility>
#include <vector>
diff --git a/funasr/runtime/onnxruntime/src/paraformer_onnx.cpp b/funasr/runtime/onnxruntime/src/paraformer_onnx.cpp
index 16cf57a..1e4a310 100644
--- a/funasr/runtime/onnxruntime/src/paraformer_onnx.cpp
+++ b/funasr/runtime/onnxruntime/src/paraformer_onnx.cpp
@@ -35,12 +35,12 @@
fbank_opts.frame_opts.frame_length_ms = 25;
fbank_opts.energy_floor = 0;
fbank_opts.mel_opts.debug_mel = false;
- //fbank_ = std::make_unique<knf::OnlineFbank>(fbank_opts);
+ // fbank_ = std::make_unique<knf::OnlineFbank>(fbank_opts);
- //sessionOptions.SetInterOpNumThreads(1);
- //sessionOptions.SetGraphOptimizationLevel(GraphOptimizationLevel::ORT_ENABLE_EXTENDED);
+ // sessionOptions.SetInterOpNumThreads(1);
sessionOptions.SetIntraOpNumThreads(nNumThread);
sessionOptions.SetGraphOptimizationLevel(ORT_ENABLE_ALL);
+ // DisableCpuMemArena can improve performance
sessionOptions.DisableCpuMemArena();
#ifdef _WIN32
--
Gitblit v1.9.1