forked from mindspore-Ecosystem/mindspore
code_docs_fix_gradreducer
This commit is contained in:
parent
18640dd4ac
commit
376120267c
|
@ -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…
Reference in New Issue