p83571920
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

modify adapter format type

This commit is contained in:
changzherui 2021-09-06 21:27:20 +08:00
parent 1b4f0bd033
commit 9169c205bd
3 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/ccsrc/transform/graph_ir/op_adapter_util.cc
View File

@ -306,7 +306,12 @@ std::string GetOpIOFormat(const AnfNodePtr &anf) {
if (iter->second == "format") {
ValuePtr format = prim->GetAttr("format");
MS_EXCEPTION_IF_NULL(format);
if (format->isa<Int64Imm>()) {
CheckAndConvertUtils::ConvertAttrValueToString(prim->name(), "format", &format);
return GetValue<std::string>(format);
} else {
return GetValue<std::string>(format);
}
}
return iter->second;
}

2
model_zoo/official/nlp/transformer/README.md
View File

@ -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,

2
model_zoo/official/nlp/transformer/README_CN.md
View File

@ -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为必填项，