Halaman utama Jelajahi Bantuan
Daftar Masuk
siweilian
/
FunASR
3
0
Fork 0
Berkas Masalah 0 Permintaan Tarik 0 Wiki
Pohon: 5e2825f0ba
Ranting Tag
FunASR
/
funasr
/
export
游雁 7114c13aa3 export model 3 tahun lalu
..
models 8bf1a6adbb exoprt model 3 tahun lalu
utils 5b0047bf58 export model 3 tahun lalu
README.md 8bf1a6adbb exoprt model 3 tahun lalu
__init__.py 59f184a622 export model 3 tahun lalu
export_model.py 7114c13aa3 export model 3 tahun lalu
test_onnx.py d567f1922c export model test 3 tahun lalu
test_torchscripts.py d567f1922c export model test 3 tahun lalu

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')