FunASR/funasr/export/README.md
2023-02-16 12:50:43 +08:00

44 lines
1.4 KiB
Markdown

## 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](../../README.md)
## Export model
`Tips`: torch 1.11.0 is required.
```shell
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
```shell
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`.
```shell
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
```shell
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`.
```shell
python -m funasr.export.export_model '/mnt/workspace/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch' "./export" false
```