INNER CODE UNIT · Python

pairwise_cosine_distance

openfoodfacts/openfoodfacts-ai · logo-ann/benchmarks/embedding_models_benchmark/main.py:127

def pairwise_cosine_distance(A: torch.Tensor) -> torch.Tensor:
    assert len(A.shape) == 2
    normalized = torch.nn.functional.normalize(A, p=2.0, dim=1)
    return 1 - torch.matmul(normalized, normalized.T)


def run_model(
    root_dir: Path,
    split_set: Optional[Set[str]],
    model_name: str,
    batch_size: int,
    device: torch.device,
) -> Tuple[torch.Tensor, torch.Tensor, LogoDataset, float]:
    model = timm.create_model(model_name, pretrained=True, num_classes=0)
    model.to(device)
    config = resolve_data_config({}, model=model)
    transform_func = create_transform(**config)
    dataset = LogoDataset(

View source record →