ColossalAI/colossalai/zero
HELSON a7d95b7024
[example] add zero1, zero2 example in GPT examples (#2146)
* [example] add zero1 and zero2 for GPT

* update readme in gpt example

* polish code

* change init value

* update readme
2022-12-20 14:30:27 +08:00
..
init_ctx
shard_utils [gemini] add GeminiMemoryManger (#832) 2022-04-24 13:08:48 +08:00
sharded_model
sharded_optim [example] add zero1, zero2 example in GPT examples (#2146) 2022-12-20 14:30:27 +08:00
sharded_param
utils
__init__.py