search

K-bNd / DINOv1_implem

DINOv1 implementation in Pytorch
0 stars 0 forks source link

Attention head visualization for ViT #10

Open K-bNd opened 1 month ago

K-bNd commented 1 month ago

Add attention head visualization in evaluation pipeline

K-bNd commented 4 weeks ago

Check out: https://pytorch.org/vision/master/feature_extraction.html

K-bNd commented 2 weeks ago

PR done for timm package https://github.com/huggingface/pytorch-image-models/pull/2220

K-bNd commented 4 days ago

PR closed, refer to this wrapper: https://github.com/huggingface/pytorch-image-models/discussions/1232