2.4K Star 8.2K Fork 4.4K

GVPMindSpore / mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.nn.PipelineGradReducer.rst 657 Bytes
一键复制 编辑 原始数据 按行查看 历史
俞涵 提交于 2024-02-27 12:05 . [i] modify format

mindspore.nn.PipelineGradReducer

.. py:class:: mindspore.nn.PipelineGradReducer(parameters, scale_sense=1.0)

    用于流水线并行的梯度聚合。

    参数:
        - **parameters** (list) - 需要更新的参数。
        - **scale_sense** (float) - 梯度的放缩系数,默认为 ``1.0``。

    异常:
        - **RuntimeError** - 如果当前模式不是图模式。
        - **RuntimeError** - 如果并行模式不是半自动并行或者自动并行。

    样例:

    .. note::
        .. include:: ../ops/mindspore.ops.comm_note.rst

        该样例需要在多卡环境下运行。

Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
r2.3

搜索帮助