Strona główna Odkrywaj Pomoc
Zarejestruj się Zaloguj się
siweilian
/
FunASR
3
0
Forkuj 0
Pliki Problemy 0 Oczekujące zmiany 0 Wiki
Drzewo: aaa5b4793d
Gałęzie Tagi
FunASR
/
funasr
/
export
游雁 7114c13aa3 export model 3 lat temu
..
models 8bf1a6adbb exoprt model 3 lat temu
utils 5b0047bf58 export model 3 lat temu
README.md 8bf1a6adbb exoprt model 3 lat temu
__init__.py 59f184a622 export model 3 lat temu
export_model.py 7114c13aa3 export model 3 lat temu
test_onnx.py d567f1922c export model test 3 lat temu
test_torchscripts.py d567f1922c export model test 3 lat temu

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