search

hpcaitech / ColossalAI-Examples

Examples of training models with hybrid parallelism using ColossalAI
Apache License 2.0
334 stars 102 forks source link

[refactor] polish pipeline example #62

Closed ver217 closed 2 years ago

ver217 commented 2 years ago

Current pipeline example run 60 epochs, which is too long for CI. I decrease the number of epochs.