visbench-classification-dinov2_vitb14

classification probe for dinov2_vitb14

A trained probe head, not a backbone. It is the small module VisBench fits on top of frozen dinov2_vitb14 features to measure what those features carry.

import visbench
from visbench.hub import load_probe_from_hub

backbone = visbench.get_backbone("dinov2_vitb14")
probe = load_probe_from_hub("turhancan97/visbench-classification-dinov2_vitb14", backbone=backbone)

It only works with this backbone

These weights were fitted on features from dinov2_vitb14, taken with pooling=cls and feature_mode=dense_only. Loading them against anything else is refused, because the failure is otherwise silent: a head fitted on one pooling and fed another has the right shapes and produces a plausible, wrong number.

Reported scores

Reproducing it

Fitted with:

• batch_size: 256
• epochs: 200
• lr: 0.01
• num_classes: 10
• optimizer: adamw
• standardize: False
• weight_decay: 0.0001

Generated by VisBench.