mirror of https://github.com/hpcaitech/ColossalAI
2731531bc2
* [autoparallel] integrate device mesh initialization into autoparallelize * add megatron solution * update gpt autoparallel examples with latest api * adapt beta value to fit the current computation cost |
||
|---|---|---|
| .. | ||
| solution_1_layers.pt | ||
| solution_4_layers.pt | ||
| solution_12_layers.pt | ||