siweilian/FunASR @ 4daea3711063c64485be3c00eaa9727404549f51

游雁 4daea37110 onnx		vor 3 Jahren
..
models	4daea37110 onnx	vor 3 Jahren
utils	5b0047bf58 export model	vor 3 Jahren
README.md	6eea85087e readme	vor 3 Jahren
__init__.py	59f184a622 export model	vor 3 Jahren
export_model.py	4daea37110 onnx	vor 3 Jahren
test_onnx.py	d567f1922c export model test	vor 3 Jahren
test_torchscripts.py	d567f1922c export model test	vor 3 Jahren

Environments

funasr 0.1.7
python 3.7
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 is required.

   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