From 930fe72f43524b4e355ef671c7180cc6cf9eefb5 Mon Sep 17 00:00:00 2001 From: 雾聪 <wucong.lyb@alibaba-inc.com> Date: 星期一, 01 四月 2024 15:27:20 +0800 Subject: [PATCH] set batch default value --- runtime/onnxruntime/include/com-define.h | 7 +++++-- 1 files changed, 5 insertions(+), 2 deletions(-) diff --git a/runtime/onnxruntime/include/com-define.h b/runtime/onnxruntime/include/com-define.h index 70e1495..9480b0b 100644 --- a/runtime/onnxruntime/include/com-define.h +++ b/runtime/onnxruntime/include/com-define.h @@ -51,8 +51,11 @@ #define QUANT_MODEL_NAME "model_quant.onnx" // gpu models #define INFER_GPU "gpu" -#define TORCH_MODEL_NAME "model.gpu.torchscripts" -#define TORCH_QUANT_MODEL_NAME "model.blade.fp16.pt" +#define BATCHSIZE "batch-size" +#define TORCH_MODEL_NAME "model.torchscripts" +#define TORCH_QUANT_MODEL_NAME "model_quant.torchscripts" +#define BLADE_MODEL_NAME "model.blade.fp16.pt" +#define BLADEDISC "bladedisc" #define VAD_CMVN_NAME "vad.mvn" #define VAD_CONFIG_NAME "vad.yaml" -- Gitblit v1.9.1