speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
f5aa97f7bf
FunASR/funasr/export
History
shixian.shi f099a04b1c export bug fix
2023-03-10 15:19:33 +08:00
..
models export bug fix 2023-03-10 15:19:33 +08:00
utils export model 2023-02-07 17:45:59 +08:00
__init__.py export model 2023-02-07 15:19:18 +08:00
export_model.py onnx export 2023-02-27 14:42:13 +08:00
README.md Update README.md 2023-03-02 13:32:41 +08:00
test_onnx.py export model test 2023-02-07 22:57:08 +08:00
test_torchscripts.py torchscripts 2023-03-02 20:20:44 +08:00

README.md

Environments

torch >= 1.11.0
modelscope >= 1.2.0

Install modelscope and funasr

The installation is the same as funasr

Export model

Tips: torch>=1.11.0

python -m funasr.export.export_model [model_name] [export_dir] [onnx]

model_name: the model is to export. It could be the models from modelscope, or local finetuned model(named: model.pb). export_dir: the dir where the onnx is export. onnx: true, export onnx format model; false, export torchscripts format model.

For example

Export onnx format model

Export model from modelscope

python -m funasr.export.export_model 'damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch' "./export" true

Export model from local path, the model'name must be model.pb.

python -m funasr.export.export_model '/mnt/workspace/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch' "./export" true

Export torchscripts format model

Export model from modelscope

python -m funasr.export.export_model 'damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch' "./export" false

Export model from local path, the model'name must be model.pb.

python -m funasr.export.export_model '/mnt/workspace/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch' "./export" false