Skip to content

Model checkpoints #15

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

Open
namin-an opened this issue Jun 25, 2024 · 4 comments
Open

Model checkpoints #15

namin-an opened this issue Jun 25, 2024 · 4 comments

Comments

@namin-an
Copy link

Hello,

Are the checkpoints (e.g., CLOOB - ResNet50) that you provided the models pretrained on CC or YFCC?
When I ran your demo notebook, I got the zero-shot classification performance of 3.66% for CLOOB - ResNet50, which seems to be similar to results in Table 1 (3.06, CLOOB trained on CC) than Table 2 (28.9 CLOOB trained on YFCC).

@fuersta
Copy link
Member

fuersta commented Jun 25, 2024

Hello,
the checkpoints are trained on YFCC. You should get the correct performance.
Your results seem to be wrong. Did you try the same for other zeroshot datasets? What are the results?

@namin-an
Copy link
Author

image
image
image

Hello,

Thank you for your reply.

The attached images are the code that I used to evaluate your pre-trained model (CLOOB—ResNet50) on the Birdsnap test dataset (1,855 images). I have also tried the Country211 dataset, which gave me lower results than the one reported in the paper.

Could you please go through my code and see if there are mistakes that I made? :(
I just changed the file name from clip to clip_cloob and downloaded the model checkpoint (https://ml.jku.at/research/CLOOB/downloads/checkpoints/cloob_rn50_yfcc_epoch_28.pt) and placed it under my checkpoint directory.

@namin-an
Copy link
Author

BTW, I got the same score (18.8) as the one in the paper for CIFAR100.

@fuersta
Copy link
Member

fuersta commented Jul 1, 2024

Did you try cloning the repo again and without changing anything running the notebook again?
I did the experiments with a clean repo and could get the same results as stated in the paper.

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