github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-14 11:23:15 +00:00
Code Issues Packages Projects Releases Wiki Activity
182 Commits 49 Branches 49 Tags 233 MiB
3af13a2c3e
Commit Graph

5 Commits

Author SHA1 Message Date
Jiarui Fang
3af13a2c3e [zero] polish ShardedOptimV2 unittest (#385) 
* place params on cpu after zero init context

* polish code

* bucketzed cpu gpu tensor transter

* find a bug in sharded optim unittest

* add offload unittest for ShardedOptimV2.

* polish code and make it more robust
 2022-03-11 15:50:28 +08:00
ver217
d0ae0f2215 [zero] update sharded optim v2 (#334) 2022-03-11 15:50:28 +08:00
ver217
1388671699 [zero] Update sharded model v2 using sharded param v2 (#323) 2022-03-11 15:50:28 +08:00
ver217
36f9a74ab2 fix sharded param hook and unit test 2022-03-11 15:50:28 +08:00
ver217
001ca624dd impl shard optim v2 and add unit test 2022-03-11 15:50:28 +08:00