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