forked from huawei/mindspore2022
modify pagedisplay
This commit is contained in:
parent
a3e771a073
commit
8dcfafda1a
|
|
@ -393,7 +393,7 @@ Boost能够自动加速网络,如减少BN/梯度冻结/累积梯度等。
|
|||
- **train_parameter_groups** (Union[tuple, list]) – 梯度冻结训练的权重。
|
||||
- **train_strategy** (Union[tuple(int), list(int), Tensor]) – 梯度冻结训练的策略。
|
||||
|
||||
.. py:function:: freeze_cell(reducer_flag, network, optimizer, sens, grad, use_grad_accumulation, mean=None, degree=None, max_accumulation_step=1)
|
||||
.. py:function:: mindspore.boost.freeze_cell(reducer_flag, network, optimizer, sens, grad, use_grad_accumulation, mean=None, degree=None, max_accumulation_step=1)
|
||||
|
||||
提供带梯度冻结的网络Cell。
|
||||
|
||||
|
|
|
|||
|
|
@ -1,4 +1,4 @@
|
|||
.. py:class:: mindspore.train.callback.History()
|
||||
.. py:class:: mindspore.train.callback.History(has_trained_epoch=0)
|
||||
|
||||
将网络输出的相关信息记录到 `History` 对象中。
|
||||
|
||||
|
|
|
|||
|
|
@ -190,12 +190,12 @@ def get_rank(group=GlobalComm.WORLD_COMM_GROUP):
|
|||
TypeError: If group is not a string.
|
||||
ValueError: If backend is invalid.
|
||||
RuntimeError: If HCCL/NCCL is not available.
|
||||
|
||||
>>> from mindspore.communication import init, get_rank
|
||||
>>> init()
|
||||
>>> rank_id = get_rank()
|
||||
>>> print(rank_id)
|
||||
>>> # the result is the rank_id in world_group
|
||||
Examples:
|
||||
>>> from mindspore.communication import init, get_rank
|
||||
>>> init()
|
||||
>>> rank_id = get_rank()
|
||||
>>> print(rank_id)
|
||||
>>> # the result is the rank_id in world_group
|
||||
"""
|
||||
if not isinstance(group, str):
|
||||
raise TypeError("For 'get_rank', the argument 'group' must be type of string, "
|
||||
|
|
|
|||
Loading…
Reference in New Issue