From cc10d01fd7d2ac51a890550d6dd67109cf7cd069 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=9B=BE=E8=81=AA?= Date: Fri, 29 Mar 2024 17:01:28 +0800 Subject: [PATCH] fix func SetBatchSize --- runtime/onnxruntime/src/paraformer-torch.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/runtime/onnxruntime/src/paraformer-torch.h b/runtime/onnxruntime/src/paraformer-torch.h index 60f15822b..e49094dc5 100644 --- a/runtime/onnxruntime/src/paraformer-torch.h +++ b/runtime/onnxruntime/src/paraformer-torch.h @@ -55,7 +55,7 @@ namespace funasr { string Rescoring(); string GetLang(){return language;}; int GetAsrSampleRate() { return asr_sample_rate; }; - void SetBatchSize(int batch_size) {batch_size_ = batch_size}; + void SetBatchSize(int batch_size) {batch_size_ = batch_size;}; int GetBatchSize() {return batch_size_;}; void StartUtterance(); void EndUtterance();