yankf21
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!9497 mode_cell_notes 

From: @bai-yangfan
Reviewed-by: @zhunaipan,@liangchenghui
Signed-off-by: @liangchenghui
This commit is contained in:
mindspore-ci-bot 2020-12-05 10:36:21 +08:00 committed by Gitee
parent a13f9acb79 306e2c6ffa
commit a8550a035f
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: