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

Will there be code release for fine tuning or training? #3

Closed
sanmeow opened this issue Jan 6, 2021 · 3 comments
Closed

Will there be code release for fine tuning or training? #3

sanmeow opened this issue Jan 6, 2021 · 3 comments

Comments

@sanmeow
Copy link

sanmeow commented Jan 6, 2021

Hi, this is a super great work!
Are there going to be anyway to fine tune this model for personal datasets?

@jongwook
Copy link
Collaborator

jongwook commented Jan 7, 2021

Thanks!

Fine-tuning should be possible with the released model. If you do the forward pass without the torch.no_grad() block, the resulting logits will have the gradients required for fine-tuning. You'll want to use the softmax cross entropy with the correct pairing (i.e. the identity matrix) as the target.

@sanmeow
Copy link
Author

sanmeow commented Jan 7, 2021

Thank!I will try it out!

@sanmeow sanmeow closed this as completed Jan 7, 2021
@launchauto
Copy link

Thank!I will try it out!

Do you success?

or-toledano added a commit to or-toledano/CLIP that referenced this issue Aug 9, 2021
jongwook pushed a commit to or-toledano/CLIP that referenced this issue Aug 9, 2021
jongwook added a commit that referenced this issue Aug 9, 2021
* Can specify root directory when loading model

* specifying download_root instead

* Update Prompt_Engineering_for_ImageNet.ipynb

Fix bug caused by changing default to jit=False with handling the case jit=True as well

* Reduce size of diff

* Reduce size of diff #2

* Reduce size of diff #3

* updated Interacting_with_CLIP.ipynb

* update Prompt_Engineering_for_ImageNet.ipynb

Co-authored-by: kcosta42 <kcosta@student.42.fr>
Co-authored-by: Jong Wook Kim <jongwook@openai.com>
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

3 participants