examples/tutorial/auto_parallel/demo_gpt2_medium.py · d5c5bc219e22e0878a14208bc963b84d969e61f4 · OpenDAS / ColossalAI

[SC] add GPT example for auto checkpoint (#1889) · d5c5bc21

Boyuan Yao authored Nov 11, 2022

* [sc] SC tutorial for auto checkpoint

* [sc] polish examples

* [sc] polish readme

* [sc] polish readme and help information

* [sc] polish readme and help information

d5c5bc21

demo_gpt2_medium.py 4.64 KB

Replace demo_gpt2_medium.py