From 63022c4f4bf064c7525c94a6ba7e62be9f3b0f94 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=BB=91cyc=E5=AE=87?= <1251519599@qq.com> Date: Thu, 12 Jan 2023 08:37:49 +0000 Subject: [PATCH] update ACL_TensorFlow/contrib/nlp/Transformer_for_ACL/README.md. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: 黑cyc宇 <1251519599@qq.com> --- .../contrib/nlp/Transformer_for_ACL/README.md | 24 +++++++------------ 1 file changed, 9 insertions(+), 15 deletions(-) diff --git a/ACL_TensorFlow/contrib/nlp/Transformer_for_ACL/README.md b/ACL_TensorFlow/contrib/nlp/Transformer_for_ACL/README.md index 00e7aaf42..fd29c31b9 100644 --- a/ACL_TensorFlow/contrib/nlp/Transformer_for_ACL/README.md +++ b/ACL_TensorFlow/contrib/nlp/Transformer_for_ACL/README.md @@ -66,21 +66,15 @@ python3 xnlp_fmk.py \ **Pb模型转换为om模型** ```Bash -python3 xnlp_fmk.py \ - --output_dir=./save/model \ - --vocab_file=./config/translate_ende_wmt32k/vocab.translate_ende_wmt32k.32768.subwords \ - --pb_model_file=./save/model/TRANSFORMER_TRANSLATION_BatchSize_1.pb \ - --om_model_file=./save/model/TRANSFORMER_TRANSLATION_BatchSize_1.om \ - --predict_batch_size=1 \ - --soc_version="Ascend310" \ - --in_nodes="\"input:1,128\"" \ - --out_nodes="\"transformer/strided_slice_11:0\"" \ - --precision_mode="allow_mix_precision" \ - --op_select_implmode="high_precision" \ - --model_name=transformer \ - --task_name=translation \ - --action_type=atc - +atc + --framework 3 \ + --soc_version "Ascend310" \ + --model ./save/model/TRANSFORMER_TRANSLATION_BatchSize_1.pb \ + --input_shape "input:1,128" \ + --output ./save/model/TRANSFORMER_TRANSLATION_BatchSize_1 \ + --out_nodes "transformer/strided_slice_11:0" \ + --op_select_implmode "high_precision" \ + --precision_mode "allow_mix_precision" ``` **运行推理** -- Gitee