facebookresearch / jepa

PyTorch code and models for V-JEPA self-supervised learning from video.
Other
2.68k stars 254 forks source link

numpy import error #40

Open tomarvimal opened 8 months ago

tomarvimal commented 8 months ago

cannot run the v-jepa attentive probe checkpoint for my custom task, Seems to be some issue with the environment . Screenshot from 2024-03-11 15-26-19

This error is a numpy import error. On further investigation i found that the environment that i am using "jepa" does not have it's numpy so python is looking for numpy in other conda env by name of "mmaction",which does have numpy installed but is failing to import above module.can someone please explain the issue and possibly any fix for the same. PS: i cloned the repo and then installed ran 'pip install -r requirements.txt'