facebookresearch/multimodal

ModuleNotFoundError: No module named 'datasets'

gihanpanapitiya opened this issue · 2 comments

Issue description

When trying to run torchmultimodal/examples/flava/train.py, I get the follwing error,

ModuleNotFoundError: No module named 'datasets'.

torchmultimodal/torchmultimodal/datasets seems to be empty.

Issue description

When trying to run torchmultimodal/examples/flava/train.py, I get the follwing error,

ModuleNotFoundError: No module named 'datasets'.

torchmultimodal/torchmultimodal/datasets seems to be empty.

I think I was missing huggingface's datasets library.

Thanks for filing the issue. Yes, its the hugging face datasets that you need. There is an open PR with the requirements.txt #9
This code is WIP for now, the author will be pushing a readme soon which will give instructions on how to run the code.