ColossalAI/colossalai/zero
ver217 3c9cd5bb5e
[zero] stateful tensor manager (#687)
* [WIP] stateful tensor manager

* add eviction strategy

* polish code

* polish code

* polish comment

* add unit test

* fix sampler bug

* polish code

* fix max sampling cnt resetting bug

* fix sampler bug

* polish code

* fix bug

* fix unit test

Co-authored-by: jiaruifang <fangjiarui123@gmail.com>
2022-04-08 17:51:34 +08:00
..
init_ctx [zero] fix init bugs in zero context (#686) 2022-04-07 17:38:45 +08:00
shard_utils [zero] stateful tensor manager (#687) 2022-04-08 17:51:34 +08:00
sharded_model [zero] stateful tensor manager (#687) 2022-04-08 17:51:34 +08:00
sharded_optim [zero] stateful tensor manager (#687) 2022-04-08 17:51:34 +08:00
sharded_param [zero] stateful tensor manager (#687) 2022-04-08 17:51:34 +08:00
__init__.py [refactor] remove old zero code (#517) 2022-03-25 14:54:39 +08:00