From 9a9b474e7de7cc90d2ee124dc8d6c2cfa887c059 Mon Sep 17 00:00:00 2001
From: xiaowan0322 <wanchen.swc@alibaba-inc.com>
Date: 星期四, 06 六月 2024 15:59:56 +0800
Subject: [PATCH] [Optimization] support bladedisc fp16 optimization (#1790)
---
examples/industrial_data_pretraining/ct_transformer/export.sh | 8 +++++---
1 files changed, 5 insertions(+), 3 deletions(-)
diff --git a/examples/industrial_data_pretraining/ct_transformer/export.sh b/examples/industrial_data_pretraining/ct_transformer/export.sh
index 7556458..780bcad 100644
--- a/examples/industrial_data_pretraining/ct_transformer/export.sh
+++ b/examples/industrial_data_pretraining/ct_transformer/export.sh
@@ -6,13 +6,14 @@
model="iic/punc_ct-transformer_zh-cn-common-vocab272727-pytorch"
-model_revision="v2.0.4"
+
python -m funasr.bin.export \
++model=${model} \
++model_revision=${model_revision} \
++type="onnx" \
-++quantize=false
+++quantize=false \
+++device="cpu"
# method2, inference from local path
@@ -21,4 +22,5 @@
python -m funasr.bin.export \
++model=${model} \
++type="onnx" \
-++quantize=false
\ No newline at end of file
+++quantize=false \
+++device="cpu"
\ No newline at end of file
--
Gitblit v1.9.1