github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-22 21:24:31 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
44b6f8947bdfd00805fbee7e121aa9def4cabee8
ColossalAI/colossalai/zero/utils/zero_hook.py
ver217 c4d903e64a [gemini] accelerate adjust_layout() (#878) 
* add lru cache

* polish code

* update unit test

* fix sharded optim
2022-04-26 18:08:31 +08:00

4.6 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink