"...en/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "dd07b19e27b737d844f62a8107228591f8d7bca8"
Unverified Commit e04e4016 authored by Samyam Rajbhandari's avatar Samyam Rajbhandari Committed by GitHub
Browse files

adding overlap_comm documentation (#247)

parent b1ddea7f
......@@ -175,6 +175,7 @@ Enabling and configure ZeRO memory optimizations
"stage": [0|1|2],
"allgather_partitions": [true|false],
"allgather_bucket_size": 500000000,
"overlap_comm": false,
"reduce_scatter": [true|false],
"reduce_bucket_size": 500000000,
"contiguous_gradients" : [true|false]
......@@ -205,6 +206,12 @@ Enabling and configure ZeRO memory optimizations
| ------------------------------------------------------------ | ------- |
| Number of elements allgathered at a time. Limits the memory required for the allgather for large model sizes | `500000000` |
***overlap_comm***: [boolean]
| Description | Default |
| ------------------------------------------------------------ | ------- |
| Attempts to overlap the reduction of the gradients with backward computation | `false` |
***reduce_scatter***: [boolean]
| Description | Default |
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment