Jiarui Fang eb1b89908c [refactor] moving InsertPostInitMethodToModuleSubClasses to utils. (#824)		2022-04-21 16:03:18 +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	[hotfix] fix memory leak in zero (#781)	2022-04-18 13:57:03 +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