ImageRetrieval

ImageRetrieval#

class stable_pretraining.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)[source]#

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[source]#

Called when the val epoch ends.

on_validation_epoch_start(trainer: Trainer, pl_module: LightningModule) → None[source]#

Called when the val epoch begins.

property state_key: str#

Unique identifier for this callback’s state during checkpointing.

Contents