search

birkhoffkiki / GPFM

The official implementation of GPFM
39 stars 1 forks source link

Model weights #2

Closed rbareja25 closed 2 months ago

rbareja25 commented 2 months ago

Hello,

I am unable to locate the model weights for this? Could you assist?

Rohan

birkhoffkiki commented 2 months ago

you could download weights from Releases

rbareja25 commented 2 months ago

Also is it a vit-large based model?Is it compatible with https://github.com/facebookresearch/dino/blob/7c446df5b9f45747937fb0d72314eb9f7b66930a/vision_transformer.py#L236

birkhoffkiki commented 2 months ago

the configuration is located at https://github.com/birkhoffkiki/GPFM/blob/2f93ebccd5ca0fe85397bf328372909de65eaa7a/models/dinov2/__init__.py#L32. I also update the readme file, see https://github.com/birkhoffkiki/GPFM/tree/master?tab=readme-ov-file#only-using-feature-extractor