ver217 9ec1ce6ab1 [zero] sharded model support the reuse of fp16 shard (#495) ...

* sharded model supports reuse fp16 shard

* rename variable

* polish code

* polish code

* polish code

2022-03-23 14:59:59 +08:00

..

bookkeeping

Feature/zero (#279)

2022-03-11 15:50:28 +08:00

__init__.py

impl shard optim v2 and add unit test

2022-03-11 15:50:28 +08:00

_utils.py

Feature/zero (#279)

2022-03-11 15:50:28 +08:00

sharded_optim_v2.py

[zero] sharded model support the reuse of fp16 shard (#495)

2022-03-23 14:59:59 +08:00

sharded_optim.py

[fp16] refactored fp16 optimizer (#392)

2022-03-15 10:05:38 +08:00