github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-13 03:38:19 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
0db43fa995a8070e69cfce3038dcdcb9a13ba90a
ColossalAI/colossalai/zero/init_ctx
History
Jiarui Fang 44e4891f57 [zero] able to place params on cpu after zero init context (#365) 
* place params on cpu after zero init context

* polish code
2022-03-11 15:50:28 +08:00
..
__init__.py
[zero] zero init context (#321)
2022-03-11 15:50:28 +08:00
init_context.py
[zero] able to place params on cpu after zero init context (#365)
2022-03-11 15:50:28 +08:00