search

hpcaitech / ColossalAI-Examples

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

added engine and trainer script for amp example #15

Closed FrankLeeeee closed 2 years ago