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 vit data parallel example #65

Closed ver217 closed 2 years ago

ver217 commented 2 years ago

Reduce the number of epochs to fit testing.