Files
ColossalAI/tests/test_zero_data_parallel/common.py
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

5.3 KiB