xla
Minor fix: ZeRO-1 grad clipping
#4796
Merged

Minor fix: ZeRO-1 grad clipping #4796

alanwaketan merged 2 commits into pytorch:master from hgt312:fix_zero1
hgt312
hgt312
JackCaoG
JackCaoG JackCaoG requested a review from alanwaketan alanwaketan 3 years ago
JackCaoG JackCaoG requested a review from AlexWertheim AlexWertheim 3 years ago
JackCaoG JackCaoG added fsdp
alanwaketan
hgt312
alanwaketan
hgt312 init
9221b721
hgt312 fix
168e44d9
hgt312 hgt312 force pushed from 357dc0cb to 168e44d9 3 years ago
alanwaketan
alanwaketan approved these changes on 2023-05-02
alanwaketan alanwaketan merged 71f9a350 into master 3 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone