游雁 d4b683433a export model 3 năm trước cách đây
..
models 8bf1a6adbb exoprt model 3 năm trước cách đây
utils 5b0047bf58 export model 3 năm trước cách đây
README.md 8bf1a6adbb exoprt model 3 năm trước cách đây
__init__.py 59f184a622 export model 3 năm trước cách đây
export_model.py d4b683433a export model 3 năm trước cách đây
test_onnx.py d567f1922c export model test 3 năm trước cách đây
test_torchscripts.py d567f1922c export model test 3 năm trước cách đây

README.md

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 onnx format model

Export model from modelscope

from funasr.export.export_model import ASRModelExportParaformer

output_dir = "../export"  # onnx/torchscripts model save path
export_model = ASRModelExportParaformer(cache_dir=output_dir, onnx=True)
export_model.export('damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')

Export model from local path

export_model.export('/root/cache/export/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')

Export torchscripts format model

Export model from modelscope

from funasr.export.export_model import ASRModelExportParaformer

output_dir = "../export"  # onnx/torchscripts model save path
export_model = ASRModelExportParaformer(cache_dir=output_dir, onnx=False)
export_model.export('damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')

Export model from local path


export_model.export('/root/cache/export/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')