github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-12 15:14:55 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
bef05489b642385c80e59fe757d598efd1752ecf
ColossalAI/colossalai/zero/init_ctx
History
Jiarui Fang 2f95df5934 [zero] able to place params on cpu after zero init context (#365) 
* place params on cpu after zero init context

* polish code
2022-03-10 14:08:58 +08:00
..
__init__.py
[zero] zero init context (#321)
2022-03-07 16:14:40 +08:00
init_context.py
[zero] able to place params on cpu after zero init context (#365)
2022-03-10 14:08:58 +08:00