Frank Lee 14e5b11d7f [zero] fixed api consistency (#1098)

2022-06-10 16:59:59 +08:00

..

init_ctx

[refactory] add nn.parallel module (#1068)

2022-06-06 15:34:41 +08:00

shard_utils

[gemini] add GeminiMemoryManger (#832)

2022-04-24 13:08:48 +08:00

sharded_model

fix import error in sharded model v2 (#1053)

2022-06-02 13:48:22 +08:00

sharded_optim

[gemini] accelerate adjust_layout() (#878)

2022-04-26 18:08:31 +08:00

sharded_param

[gemini] add GeminiMemoryManger (#832)

2022-04-24 13:08:48 +08:00

utils

[gemini] zero supports gemini (#1093)

2022-06-10 14:48:28 +08:00

__init__.py

[zero] add zero optimizer for ColoTensor (#1046)

2022-06-02 12:13:15 +08:00

zero_optimizer.py

[zero] fixed api consistency (#1098)

2022-06-10 16:59:59 +08:00