ColossalAI/colossalai/zero
Jiarui Fang 595bedf767
revert zero tensors back (#829)
2022-04-22 12:12:35 +08:00
..
init_ctx [refactor] moving InsertPostInitMethodToModuleSubClasses to utils. (#824) 2022-04-21 16:03:18 +08:00
shard_utils Revert "[zero] add ZeroTensorShardStrategy (#793)" (#806) 2022-04-19 14:40:02 +08:00
sharded_model [refactor] moving memtracer to gemini (#801) 2022-04-19 10:13:08 +08:00
sharded_optim [log] local throughput metrics (#811) 2022-04-20 10:05:39 +08:00
sharded_param revert zero tensors back (#829) 2022-04-22 12:12:35 +08:00
utils [gemini] collect cpu-gpu moving volume in each iteration (#813) 2022-04-20 11:29:48 +08:00
__init__.py [log] local throughput metrics (#811) 2022-04-20 10:05:39 +08:00