OpenBMB / ModelCenter

Efficient, Low-Resource, Distributed transformer implementation based on BMTrain
https://modelcenter.readthedocs.io
Apache License 2.0
233 stars 28 forks source link

[BUG] can't import model_center.tools #28

Closed xrandx closed 2 years ago

xrandx commented 2 years ago

Describe the bug After installing by pip or installing by source, I can't import model_center.tools like your doc. I look at the folder "/miniconda3/envs/***/lib/python3.9/site-packages/model_center". This folder doesn't contain tools.

Expected behavior import module directly.

Environment: Ubuntu 18.84 _pytorch_select 0.1 cpu_0 pytorch 1.11.0 py3.9_cuda11.3_cudnn8.2.0_0 pytorch-mutex 1.0 cuda torchaudio 0.11.0 py39_cu113
torchvision 0.12.0 py39_cu113

MayDomine commented 2 years ago

The bug has been fixed.You can pull latest code to avoid this issue.