Browse Source

!9497 mode_cell_notes

From: @bai-yangfan
Reviewed-by: @zhunaipan,@liangchenghui
Signed-off-by: @liangchenghui
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
a8550a035f
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      mindspore/common/parameter.py
  2. +1
    -1
      mindspore/nn/cell.py

+ 1
- 1
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")):


+ 1
- 1
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:


Loading…
Cancel
Save