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

Are there any other model weights? #15

Open
yangzhou321 opened this issue Dec 30, 2023 · 1 comment
Open

Are there any other model weights? #15

yangzhou321 opened this issue Dec 30, 2023 · 1 comment

Comments

@yangzhou321
Copy link

hi~ I just found that you mentioned in this repo that data features were extracted from some models like "ResNet50, MoCoWideResNetCOAD, iBOTViTSmallCOAD, iBOTViTBaseCOAD, iBOTViTBasePANCAN, iBOTViTLargeCOAD". I recently found that the pre-trained "phikon" (ViT-base) model you provided does not seem to perform well on my own dataset. Are there any pre-trained weights of these models? Perhaps the performance of ViT-L is better on certain datasets? Could you please provide me with some pre-trained vit-large models iBOT-ViT-Large-COAD or iBOT-ViT-Large-PANCAN? Thanks a lot.

@afilt
Copy link
Collaborator

afilt commented Jan 3, 2024

Hi @yangzhou321 !
Thanks for your feedback. Do you use the teacher backbone to extract features ? What is the task you're trying to solve if I may ?

As of now, we don't plan to release further weights on this repo (like iBOT-ViT-Large-COAD or iBOT-ViT-Large-PANCAN) but stay tuned, some may be released in the coming weeks.

Also, iBOTViTBasePANCAN (aka Phikon on Hugging-Face) is the more complete model in terms of generalization to different cancer indications; but of course you may find better results with specialized / fine-tuned models (for instance, CRC-trained models for CRC-related tasks).

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