Skip to content

enable resultcaching for ViT models from timm#2235

Merged
KartikP merged 1 commit intobrain-score:masterfrom
KogniJannis:enable-vit-resultcaching
Jan 8, 2026
Merged

enable resultcaching for ViT models from timm#2235
KartikP merged 1 commit intobrain-score:masterfrom
KogniJannis:enable-vit-resultcaching

Conversation

@KogniJannis
Copy link
Copy Markdown
Collaborator

Previously, the timm model plugin checked if a model is a ViT and if yes, disabled caching of extracted activations. This was because of the problem described in #1232 and addressed in #2164 which merged. Therefore, I think we should be able to enable result caching now.

@KartikP
Copy link
Copy Markdown
Collaborator

KartikP commented Jan 8, 2026

Thanks for this @KogniJannis this looks good to me.

Copy link
Copy Markdown
Collaborator

@KartikP KartikP left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@KartikP KartikP merged commit 30410d1 into brain-score:master Jan 8, 2026
3 checks passed
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

Successfully merging this pull request may close these issues.

2 participants