Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

ModuleNotFoundError: No module named 'clip' #4

Closed
Jiang15 opened this issue Jun 1, 2021 · 8 comments
Closed

ModuleNotFoundError: No module named 'clip' #4

Jiang15 opened this issue Jun 1, 2021 · 8 comments

Comments

@Jiang15
Copy link

Jiang15 commented Jun 1, 2021

Thanks for your great work! when I call train.py with my dataset. I have ModuleNotFoundError: No module named 'clip'. Do I miss something?

@Zasder3
Copy link
Owner

Zasder3 commented Jun 1, 2021

Sorry I haven't made a file stating the dependencies; it requires the original CLIP repository's pip installation protocol. Simply enter the following into your terminal and it will install the clip module:

$ pip install git+https://github.com/openai/CLIP.git

@Zasder3 Zasder3 closed this as completed Jun 1, 2021
@Jiang15
Copy link
Author

Jiang15 commented Jun 1, 2021 via email

@Zasder3
Copy link
Owner

Zasder3 commented Jun 1, 2021

This is an important point. Our repo starts from random inits and CLIP used 400M text image pairs. They don't even state what their accuracy is during a batch... One thing that could be helpful is using a pretrained initialization and see what happens from there. I'll make sure I double check to ensure the accuracy is correct though.

@Jiang15
Copy link
Author

Jiang15 commented Jun 2, 2021 via email

@Zasder3
Copy link
Owner

Zasder3 commented Jun 2, 2021

Looks like the accuracy calculation was incorrect! I'll update the repo.

@Jiang15
Copy link
Author

Jiang15 commented Jun 5, 2021 via email

@Zasder3
Copy link
Owner

Zasder3 commented Jun 5, 2021

Hi @Jiang15, the photo you sent over sadly isn't visible. Try using the GitHub website or app to insert it.

Although I can't see your error I do have some useful feedback on reloading the weights. Lightning requires you to actually use the checkpoints that the training run spits out. There's likely folder called lightning_Logs that will have your model saved. Checkout the documentation here and use the class CLIPWrapper to receive the checkpoint.

@Jiang15
Copy link
Author

Jiang15 commented Jun 6, 2021 via email

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants