Jiarui Fang 294a6060d0 [tensor] ZeRO use ColoTensor as the base class. (#828) ... * [refactor] moving InsertPostInitMethodToModuleSubClasses to utils. * [tensor] ZeRO use ColoTensor as the base class. * polish		2022-04-22 12:00:48 +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	[tensor] ZeRO use ColoTensor as the base class. (#828)	2022-04-22 12:00:48 +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