ImageRetrieval

class stable_ssl.callbacks.ImageRetrieval(pl_module, name: str, input: str, query_col: str, retrieval_col: str | List[str], metrics, features_dim: tuple[int] | list[int] | int, normalizer: str | None = None)

Bases: Callback

Image Retrieval evaluator for self-supervised learning.

The implementation follows:

1. facebookresearch/dino

on_validation_epoch_end(trainer: Trainer, pl_module: LightningModule) → None

Called when the val epoch ends.

on_validation_epoch_start(trainer: Trainer, pl_module: LightningModule) → None

Called when the val epoch begins.