Browse Source

code_docs_fix_gradreducer

feature/build-system-rewrite
lichenever 4 years ago
parent
commit
376120267c
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      docs/api/api_python/nn/mindspore.nn.DistributedGradReducer.rst

+ 1
- 0
docs/api/api_python/nn/mindspore.nn.DistributedGradReducer.rst View File

@@ -13,6 +13,7 @@ mindspore.nn.DistributedGradReducer
- **mean** (bool) - 当mean为True时,对AllReduce之后的梯度求均值。默认值:False。
- **degree** (int) - 平均系数,通常等于设备编号。默认值:None。
- **fusion_type** (int) - AllReduce算子的融合类型。默认值:1。
- **group** (str) - AllReduce算子的通信域,若需要自定义通信域,需要调用create_group接口。默认值:GlobalComm.WORLD_COMM_GROUP。

**异常:**



Loading…
Cancel
Save