From 7ac4fa555a38a26b0db3e240bfc6ff6831788aa1 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 27 二月 2023 19:09:32 +0800
Subject: [PATCH] torch version
---
funasr/export/README.md | 8 ++++----
1 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/funasr/export/README.md b/funasr/export/README.md
index 513bbb2..846978f 100644
--- a/funasr/export/README.md
+++ b/funasr/export/README.md
@@ -10,12 +10,12 @@
The installation is the same as [funasr](../../README.md)
## Export model
- `Tips`: torch 1.11.0 is required.
+ `Tips`: torch>=1.11.0
```shell
python -m funasr.export.export_model [model_name] [export_dir] [onnx]
```
- `model_name`: the model is to export.
+ `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.
@@ -25,7 +25,7 @@
```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
```
@@ -36,7 +36,7 @@
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