From 306e2c6ffa69768303ae081ca9ade92bf7b0a2fc Mon Sep 17 00:00:00 2001 From: bai-yangfan Date: Fri, 4 Dec 2020 14:30:26 +0800 Subject: [PATCH] mode_cell_notes --- mindspore/common/parameter.py | 2 +- mindspore/nn/cell.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/common/parameter.py b/mindspore/common/parameter.py index f35d2edc5f..0aae4e2b3f 100644 --- a/mindspore/common/parameter.py +++ b/mindspore/common/parameter.py @@ -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")): diff --git a/mindspore/nn/cell.py b/mindspore/nn/cell.py index 0c06cd2886..0b65936f87 100755 --- a/mindspore/nn/cell.py +++ b/mindspore/nn/cell.py @@ -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: