examples/language/gpt/train_gpt_demo.py · a7d95b7024454f8f79d3b9631a9862671b0dc1cc · OpenDAS / ColossalAI

HELSON authored Dec 20, 2022

* [example] add zero1 and zero2 for GPT

* update readme in gpt example

* polish code

* change init value

* update readme

a7d95b70

train_gpt_demo.py 10.9 KB