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
6c78946fdd88a1cc950deeec46119009ef522efd
ColossalAI/colossalai/zero
History
Jiarui Fang 6c78946fdd Polish sharded parameter (#297) 
* init shard param from shape tuple

* add more unitest for shard param

* add more unittests to shareded param
2022-03-03 12:42:57 +08:00
..
sharded_model
Polish sharded parameter (#297)
2022-03-03 12:42:57 +08:00
sharded_optim
Feature/zero (#279)
2022-03-01 18:17:01 +08:00
sharded_param
Polish sharded parameter (#297)
2022-03-03 12:42:57 +08:00
__init__.py
added buffer sync to naive amp model wrapper (#291)
2022-03-02 16:47:17 +08:00