Browse Source

change parameter init_data log level to debug

tags/v0.5.0-beta
Yi Huaijie 5 years ago
parent
commit
a621e22eac
2 changed files with 4 additions and 4 deletions
  1. +1
    -1
      mindspore/common/api.py
  2. +3
    -3
      mindspore/nn/cell.py

+ 1
- 1
mindspore/common/api.py View File

@@ -362,7 +362,7 @@ class _Executor:
if not auto_parallel_mode:
param.init_data()
elif key not in obj.parameter_layout_dict:
logger.info("Layout dict does not contain the key %s.", key)
logger.debug("Layout dict does not contain the key %s.", key)
param.init_data(set_sliced=True)
else:
layout = obj.parameter_layout_dict[key]


+ 3
- 3
mindspore/nn/cell.py View File

@@ -305,7 +305,7 @@ class Cell:
logger.info("layout dict does not contain the key %s", key)
continue
if self.parameters_dict()[key].sliced:
logger.info("Param %s is already sliced.", key)
logger.debug("Param %s is already sliced.", key)
continue
layout = self.parameter_layout_dict[key]
new_tensor = _load_tensor_by_layout(tensor, layout)
@@ -318,7 +318,7 @@ class Cell:
logger.info("layout dict does not contain the key %s", key)
continue
if params[key].sliced:
logger.info("Param %s is already sliced.", key)
logger.debug("Param %s is already sliced.", key)
continue
layout = self.parameter_layout_dict[key]
new_tensor = _load_tensor_by_layout(tensor, layout)
@@ -484,7 +484,7 @@ class Cell:
if not auto_parallel_mode:
param.init_data()
elif param.name not in self.parameter_layout_dict:
logger.info("Layout dict does not contain the key %s.", param.name)
logger.debug("Layout dict does not contain the key %s.", param.name)
param.init_data(set_sliced=True)
else:
layout = self.parameter_layout_dict[param.name]


Loading…
Cancel
Save