You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

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		3 years ago
..
init_ctx	[refactor] moving InsertPostInitMethodToModuleSubClasses to utils. (#824)	3 years ago
shard_utils	Revert "[zero] add ZeroTensorShardStrategy (#793)" (#806)	3 years ago
sharded_model	[refactor] moving memtracer to gemini (#801)	3 years ago
sharded_optim	[log] local throughput metrics (#811)	3 years ago
sharded_param	[tensor] ZeRO use ColoTensor as the base class. (#828)	3 years ago
utils	[gemini] collect cpu-gpu moving volume in each iteration (#813)	3 years ago
__init__.py	[log] local throughput metrics (#811)	3 years ago