hpcaitech / ColossalAI-Examples

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

change to fit refactored schedule #75

Closed YuliangLiu0306 closed 2 years ago

YuliangLiu0306 commented 2 years ago

The colossalAI repo has put schedule into engine module, so some old training codes need to be updated to fit refactored schedule.