ModuleNotFoundError: No module named 'training'

mlfoundations / datacomp

DataComp: In search of the next generation of multimodal datasets

http://datacomp.ai/

Other

588 stars 49 forks source link

ModuleNotFoundError: No module named 'training' #79

Closed adymaharana closed 3 months ago

adymaharana commented 3 months ago

Is this codebase missing a module called training or am I doing something weird? Asking because I ran this codebase several times last year and never faced an error until today when I cloned a fresh version from GitHub.

Traceback (most recent call last):
  File "/nas-ssd/adyasha/projects/datacomp/train.py", line 12, in <module>
    from training.distributed import world_info_from_env
ModuleNotFoundError: No module named 'training'

Any help is appreciated, thank you!

sagadre commented 3 months ago

hi @adymaharana! did you also install the virtual env from scratch? if not, maybe try that? training comes from open_clip, see here for more details: https://github.com/mlfoundations/datacomp/issues/32

adymaharana commented 3 months ago

Hi @sagadre ! So I was doing something weird :) My virtual environment setup had errors and I didn't notice. Thanks for catching that.