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

mode_cell_notes

This commit is contained in:
bai-yangfan 2020-12-04 14:30:26 +08:00
parent c81bb8fe39
commit 306e2c6ffa
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/common/parameter.py
View File

@ -187,7 +187,7 @@ class Parameter(MetaTensor_):
Args:
init_in_server (bool): Whether trainable parameter updated by parameter server is
initialized on server. Default: False.
initialized on server. Default: False.
"""
if _is_role_worker() or _is_role_pserver() or _is_role_sched():
if init_in_server and (not self.name.endswith("embedding_table")):

2
mindspore/nn/cell.py
View File

@ -1064,7 +1064,7 @@ class Cell(Cell_):
Args:
recurse (bool): Whether sets the trainable parameters of subcells. Default: True.
init_in_server (bool): Whether trainable parameters updated by parameter server are
initialized on server. Default: False.
initialized on server. Default: False.
"""
params = self.trainable_params(recurse)
for param in params: