NVlabs / prismer

The implementation of "Prismer: A Vision-Language Model with Multi-Task Experts".
https://shikun.io/projects/prismer
Other
1.3k stars 75 forks source link

Getting issues on running demo.py as imports have been done wrongly #16

Closed chanangad closed 1 year ago

chanangad commented 1 year ago

File "/Users/achandhok/Documents/GitHub/prismer/experts/edge/generate_dataset.py", line 10, in from dataset.utils import * ModuleNotFoundError: No module named 'dataset'

dataset folder is not in the current but parent directory

lorenmt commented 1 year ago

You should run the demo in the prismer directory. And set pythonpath=. Alternatively, I have added the pythonpath in the script, so just run python demo.py --exp_name MODEL_NAME should work.

lorenmt commented 1 year ago

No further updates.