INNER CODE UNIT · Python

squared_sum

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

    squared_sum = torch.sum(A**2.0, axis=1, keepdim=True)
    return squared_sum + squared_sum.T - 2 * dot_product


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)

View source record →