ModuleNotFoundError

[EYUJin]

Hi, I am trying to run the slot_attention/train.py code in your repository, however, I keep getting stuck by a error reading...

Traceback (most recent call last):
  File "slot_attention/train.py", line 8, in <module>
    from slot_attention.data import CLEVRDataModule
ModuleNotFoundError: No module named 'slot_attention'

I cloned the git to a PC running on the Ubuntu operating system, and upon doing so, I ran the code within the highest directory of the cloned repository, as written in the Readme. Could you help me out with this issue?

[joshalbrecht]

What command are you running? From what directory? What is your PYTHONPATH set to?

[greeneggsandyaml]

To expand on the comment above, PYTHONPATH=. python slot_attention/train.py should solve this issue.

[EYUJin]

Thanks! It seems to work now :) 2021년 6월 3일 (목) 오후 10:07, greeneggsandyaml ***@***.***>님이 작성:

…

To expand on the comment above, PYTHONPATH=. python slot_attention/train.py should solve this issue. — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub <#5 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/APF7OEDFSC5RE5IZWEM6L5DTQ547HANCNFSM43562LUQ> .

[HHousen]

You can also use python -m slot_attention.train to run the training script as a module instead of changing the Python path.