search

hpcaitech / ColossalAI-Examples

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

BERT示例运行错误，ColossalAI-Examples/language/bert/sequene_parallel/ #158

Closed ZXM1063694570 closed 1 year ago

ZXM1063694570 commented 1 year ago

🐛 Describe the bug

使用了最新提供的Dockerhub上的镜像0.1.8，但是在运行BERT序列并行案例：ColossalAI-Examples/language/bert/sequene_parallel/时候仍不能正常运行，提示缺少相关包： Traceback (most recent call last): File "/workspace/ColossalAI-Examples/language/bert/sequene_parallel/train.py", line 10, in from model.bert import BertForPretrain File "/workspace/ColossalAI-Examples/language/bert/sequene_parallel/model/bert.py", line 12, in from colossalai.builder.pipeline import partition_uniform ModuleNotFoundError: No module named 'colossalai.builder.pipeline'

XI 7HJD3ZJY0X1W2RW W}6Q

Environment

docker镜像：docker pull hpcaitech/colossalai:0.1.8

tianboh commented 1 year ago

I have met this error as well. Use from colossalai.pipeline.utils import partition_uniform instead.

ZXM1063694570 commented 1 year ago

我也遇到过这个错误。改为使用from colossalai.pipeline.utils import partition_uniform。

It worked. Thank you