From 2cdb2d654f2109ef4e648bae6f169143e267e5db Mon Sep 17 00:00:00 2001 From: zhuzizyf <42790740+zhuzizyf@users.noreply.github.com> Date: 星期六, 11 三月 2023 14:33:14 +0800 Subject: [PATCH] Update dataset.py --- funasr/export/README.md | 26 +++++++++++++++++--------- 1 files changed, 17 insertions(+), 9 deletions(-) diff --git a/funasr/export/README.md b/funasr/export/README.md index a4d61bb..c44ad33 100644 --- a/funasr/export/README.md +++ b/funasr/export/README.md @@ -1,32 +1,40 @@ ## Environments - funasr 0.1.7 - python 3.7 - torch 1.11.0 - modelscope 1.2.0 + torch >= 1.11.0 + modelscope >= 1.2.0 ## Install modelscope and funasr The installation is the same as [funasr](../../README.md) -## Export onnx format model +## Export model + `Tips`: torch>=1.11.0 + + ```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 +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 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 +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 ``` -- Gitblit v1.9.1