m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!13518 add annotation for python api and modify some log level

Browse Source 
From: @wangnan39
Reviewed-by: @kingxian,@zh_qh
Signed-off-by: @kingxian,@zh_qh
pull/13518/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
parent 5fd3d140b6 1c58e0de1a
commit e1cdd49177
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File

8
mindspore/core/utils/check_convert_utils.cc
Unescape View File

@ -230,7 +230,7 @@ bool CheckAndConvertUtils::ConvertAttrValueToInt(const std::string &op_type, con
bool CheckAndConvertUtils::ConvertAttrValueToString(const std::string &op_type, const std::string &attr_name,
ValuePtr *const value) {
if (value == nullptr || *value == nullptr) {
MS_LOG(INFO) << "value is nullptr! op_type = " << op_type << ", attr_name = " << attr_name;
MS_LOG(DEBUG) << "value is nullptr! op_type = " << op_type << ", attr_name = " << attr_name;
return false;
}
if (!(*value)->isa<Int64Imm>()) {
@ -280,7 +280,7 @@ void RestoreTargetAttr(const std::string &attr_name, ValuePtr *const value) {
void CheckAndConvertUtils::ConvertAttrValueInExport(const std::string &op_type, const std::string &attr_name,
ValuePtr *const value) {
if (value == nullptr || *value == nullptr) {
MS_LOG(INFO) << "value is nullptr! op_type = " << op_type << ", attr_name = " << attr_name;
MS_LOG(DEBUG) << "value is nullptr! op_type = " << op_type << ", attr_name = " << attr_name;
return;
}
// convert enum to string
@ -292,7 +292,7 @@ void CheckAndConvertUtils::ConvertAttrValueInExport(const std::string &op_type,
void CheckAndConvertUtils::ConvertAttrValueInLoad(const std::string &op_type, const std::string &attr_name,
ValuePtr *const value) {
if (value == nullptr || *value == nullptr) {
MS_LOG(INFO) << "value is nullptr! op_type = " << op_type << ", attr_name = " << attr_name;
MS_LOG(DEBUG) << "value is nullptr! op_type = " << op_type << ", attr_name = " << attr_name;
return;
}
// convert string to enum
@ -626,7 +626,7 @@ TypeId CheckAndConvertUtils::CheckTypeSame(const std::string &arg_name, const Ty
bool CheckAndConvertUtils::CheckIrAttrtoOpAttr(const std::string &op_type, const std::string &attr_name,
ValuePtr *const value) {
if (*value == nullptr) {
MS_LOG(INFO) << "value is nullptr! op_type = " << op_type << ", attr_name = " << attr_name;
MS_LOG(DEBUG) << "value is nullptr! op_type = " << op_type << ", attr_name = " << attr_name;
return false;
}
if (op_type.empty() || attr_name.empty()) {

3
mindspore/nn/cell.py
Unescape View File

@ -1051,7 +1051,8 @@ class Cell(Cell_):
"""
Sets the cell to training mode.
The cell itself and all children cells will be set to training mode.
The cell itself and all children cells will be set to training mode. Layers that have different constructions
for training and predicting , such as `BatchNorm`, will distinguish between the branches by this attribute.
Args:
mode (bool): Specifies whether the model is training. Default: True.

Write Preview
Loading…
Cancel
Save