diff --git a/mindspore/ccsrc/transform/graph_ir/op_adapter_util.cc b/mindspore/ccsrc/transform/graph_ir/op_adapter_util.cc index 54c4922360c..e02998da5cc 100644 --- a/mindspore/ccsrc/transform/graph_ir/op_adapter_util.cc +++ b/mindspore/ccsrc/transform/graph_ir/op_adapter_util.cc @@ -306,7 +306,12 @@ std::string GetOpIOFormat(const AnfNodePtr &anf) { if (iter->second == "format") { ValuePtr format = prim->GetAttr("format"); MS_EXCEPTION_IF_NULL(format); - return GetValue(format); + if (format->isa()) { + CheckAndConvertUtils::ConvertAttrValueToString(prim->name(), "format", &format); + return GetValue(format); + } else { + return GetValue(format); + } } return iter->second; } diff --git a/model_zoo/official/nlp/transformer/README.md b/model_zoo/official/nlp/transformer/README.md index 3a4ad80b827..5ecc158deff 100644 --- a/model_zoo/official/nlp/transformer/README.md +++ b/model_zoo/official/nlp/transformer/README.md @@ -358,7 +358,7 @@ Parameters for learning rate: ### [Export MindIR](#contents) ```shell -python export.py --ckpt_file [CKPT_PATH] --file_name [FILE_NAME] --file_format [FILE_FORMAT] +python export.py --model_file [CKPT_PATH] --file_name [FILE_NAME] --file_format [FILE_FORMAT] ``` The ckpt_file parameter is required, diff --git a/model_zoo/official/nlp/transformer/README_CN.md b/model_zoo/official/nlp/transformer/README_CN.md index f14c8435c56..44bcf179a7e 100644 --- a/model_zoo/official/nlp/transformer/README_CN.md +++ b/model_zoo/official/nlp/transformer/README_CN.md @@ -361,7 +361,7 @@ Parameters for learning rate: ### [导出MindIR](#contents) ```shell -python export.py --ckpt_file [CKPT_PATH] --file_name [FILE_NAME] --file_format [FILE_FORMAT] +python export.py --model_file [CKPT_PATH] --file_name [FILE_NAME] --file_format [FILE_FORMAT] ``` 参数ckpt_file为必填项,