Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[zero3 doc] a few small tweaks #839

Merged
merged 3 commits into from
Mar 11, 2021
Merged

Conversation

stas00
Copy link
Collaborator

@stas00 stas00 commented Mar 9, 2021

a few small tweaks

@ShadenSmith
Copy link
Contributor

Admin merging as the doc-only edits don't trigger our CI build.

@ShadenSmith ShadenSmith merged commit 7925d0c into deepspeedai:master Mar 11, 2021
jeffra added a commit to jeffra/DeepSpeed that referenced this pull request Aug 25, 2021
* set adamw_mode default true (follows FusedAdam and < 0.3.11 logic) (deepspeedai#844)

* less scary overflow notice (deepspeedai#833)

Co-authored-by: Jeff Rasley <[email protected]>

* Add optimizers and schedules to RTD and updated the corresponding part in the website (deepspeedai#799)

* add optimizers and schedules to rtd

* update ds website and fix links

* add optimizers and schedules to rtd

* update ds website and fix links

* add flops profiler to rtd

* fix

Co-authored-by: Shaden Smith <[email protected]>

* small tweaks (deepspeedai#839)

* Control ZeRO wall clock timers (deepspeedai#849)

* Control ZeRO wall clock timers

* Disable more ZeRO3 debug prints

Co-authored-by: Jeff Rasley <[email protected]>

* [WarmupDecayLR] fix log(0) & 1/log(1) bugs (deepspeedai#772)

* fix log(0) & 1/log(1) bugs

* simplify

Co-authored-by: Jeff Rasley <[email protected]>
Co-authored-by: Reza Yazdani <[email protected]>
Co-authored-by: Cheng Li <[email protected]>

* bump to v0.3.12

* Bug fix: Remove client optimizer param_group list item that does not have 'params' (deepspeedai#827)

Co-authored-by: Jeff Rasley <[email protected]>

* [doc] pipeline doc typos/improvements (deepspeedai#659)

Admin merging for pure-doc PR that does not trigger build.

* Samyamr/inference hook fix (deepspeedai#851)

* Fix mis-aligned-grad

When a parameter is not divisible by world size, the partitioned gradients are mis-aligned due to incorrect padding handling. This PR should fix for that.

* Formatting fix

* Adding static_scale test back for Z3, and also changing hidden size to be not divisile by world_size

* also removing alignment from flat fp16 buffers

* Testing for hidden dim alignment

* inference hook fix

* Update stage3.py

* formatting

* [bug-fix] move params to gpu if offload params is turned off

Co-authored-by: Samyam Rajbhandari <[email protected]>
Co-authored-by: Shaden Smith <[email protected]>
Co-authored-by: Jeff Rasley <[email protected]>

* ZeRO Stage 2: Clear reduced gradients (deepspeedai#856)

* Ensure gradients of other partitions are cleared after reduction

* Remove redundant code

Co-authored-by: Jeff Rasley <[email protected]>

* Squash stage3 v1 (deepspeedai#146)

Co-authored-by: Samyam <[email protected]>
Co-authored-by: Jeff Rasley <[email protected]>
Co-authored-by: Samyam Rajbhandari <[email protected]>
Co-authored-by: Olatunji Ruwase <[email protected]>
Co-authored-by: Shaden Smith <[email protected]>
Co-authored-by: Shaden Smith <[email protected]>
Co-authored-by: eltonzheng <[email protected]>

* formatting fix (deepspeedai#150)

* stage3 bugfix (API) update and simplified FP16 Z3 tests (deepspeedai#151)

* fp16 Z3 API update and bugfix

* revert debug change

* docs

* filling in allocation docs

* better assumption docs

* doc progress

* config json

* major docs edits

* auto registration works for accessed cases

* working on small models.

* debugging large-model discovery?

* fix discovery to first forward pass?

* return obj ext param

* support None parameters in auto-discovery

Co-authored-by: Jeff Rasley <[email protected]>
Co-authored-by: Stas Bekman <[email protected]>
Co-authored-by: Cheng Li <[email protected]>
Co-authored-by: Olatunji Ruwase <[email protected]>
Co-authored-by: Reza Yazdani <[email protected]>
Co-authored-by: Samyam Rajbhandari <[email protected]>
Co-authored-by: eltonzheng <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants