ver217 c5cd3b0f35 [zero] zero optim copy chunk rather than copy tensor (#1070)

2022-06-07 10:30:46 +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

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

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

__init__.py

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

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

zero_optimizer.py

[zero] zero optim copy chunk rather than copy tensor (#1070)

2022-06-07 10:30:46 +08:00