mode_cell_notes

pull/9497/head
bai-yangfan 4 years ago
parent c81bb8fe39
commit 306e2c6ffa

@ -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")):

@ -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