From 82e5ca37a8bd80f56c99f9d790a03b458ced716b Mon Sep 17 00:00:00 2001 From: 游雁 <zhifu.gzf@alibaba-inc.com> Date: 星期二, 25 二月 2025 14:28:34 +0800 Subject: [PATCH] Large-Scale Data Training --- export.py | 5 +++-- 1 files changed, 3 insertions(+), 2 deletions(-) diff --git a/export.py b/export.py index 3307e96..a891e60 100644 --- a/export.py +++ b/export.py @@ -2,8 +2,9 @@ from funasr import AutoModel model = AutoModel( - model="/raid/t3cv/wangch/WORK_SAPCE/ASR/models/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch" + model="iic/emotion2vec_base", + hub="ms" ) -res = model.export(type="onnx", quantize=False, opset_version=13, device='cuda') # fp32 onnx-gpu +res = model.export(type="onnx", quantize=False, opset_version=13, device='cpu') # fp32 onnx-gpu # res = model.export(type="onnx_fp16", quantize=False, opset_version=13, device='cuda') # fp16 onnx-gpu -- Gitblit v1.9.1