hpcaitech/ColossalAI-Examples

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

ZXM1063694570 opened this issue · 2 comments

ZXM1063694570 commented

🐛 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

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

ZXM1063694570 commented

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

It worked. Thank you