github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-01 20:33:02 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
c9c37dcc4dfeb58eb72cf24b9ea5cfd9b593fd99
ColossalAI/colossalai/zero/zero_optimizer.py
ver217 561e90493f [zero] zero optim supports loading local state dict (#1171) 
* zero optim supports loading local state dict

* polish code

* add unit test
2022-06-24 17:25:57 +08:00

7.5 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink