ColossalAI/colossalai/zero
ver217 ab8c6b4a0e
[zero] refactor memstats collector (#706)
* refactor memstats collector

* fix disposable

* polish code
2022-04-11 10:46:08 +08:00
..
init_ctx [zero] adapt zero hooks for unsharded module (#699) 2022-04-08 20:23:26 +08:00
shard_utils [zero] refactor memstats collector (#706) 2022-04-11 10:46:08 +08:00
sharded_model [zero] refactor memstats collector (#706) 2022-04-11 10:46:08 +08:00
sharded_optim [zero] adapt zero hooks for unsharded module (#699) 2022-04-08 20:23:26 +08:00
sharded_param [zero] adapt zero hooks for unsharded module (#699) 2022-04-08 20:23:26 +08:00
__init__.py [refactor] remove old zero code (#517) 2022-03-25 14:54:39 +08:00