Browse Source

[ME] Add deprecated warning when the user does not set a name for the parameter.

tags/v1.6.0
Margaret_wangrui 4 years ago
parent
commit
d82d26ce0f
2 changed files with 6 additions and 1 deletions
  1. +4
    -1
      mindspore/common/parameter.py
  2. +2
    -0
      mindspore/nn/cell.py

+ 4
- 1
mindspore/common/parameter.py View File

@@ -17,6 +17,7 @@
from copy import copy
import numbers
import numpy as np
from mindspore import log as logger
from .._c_expression import ParamInfo
from . import dtype as mstype
from .. import context
@@ -332,7 +333,9 @@ class Parameter(Tensor_):
raise RuntimeError("The dims of parameter '{}' must be 2, but got {}."
.format(self.name, len(self.shape)))
_reinsert_hash_table_size(name_, self.param_info.name, self.shape[0], self.shape[1])

if name_ == PARAMETER_NAME_DEFAULT:
logger.warning("The parameter definition is deprecated.\n"
"Please set a unique name for the parameter '{}'.". format(self))
self.param_info.name = name_

@property


+ 2
- 0
mindspore/nn/cell.py View File

@@ -565,6 +565,8 @@ class Cell(Cell_):
continue
exist_objs.add(item)
if item.name == PARAMETER_NAME_DEFAULT:
logger.warning("The parameter definition is deprecated.\n"
"Please set a unique name for the parameter in ParameterTuple '{}'.". format(value))
item.name = item.name + "$" + str(self._id)
self._id += 1
self.insert_param_to_cell(item.name, item, check_name_contain_dot=False)


Loading…
Cancel
Save