search

hpcaitech / ColossalAI-Examples

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

debug zero exmaple #184

Closed feifeibear closed 1 year ago

feifeibear commented 1 year ago

train_v2.py can not work under torch version 1.12.0! @1SAA

feifeibear commented 1 year ago
image