From 0df232277444af28d2c2abbd738ca5e32f5d14a2 Mon Sep 17 00:00:00 2001 From: Inga Ulusoy Date: Thu, 27 Apr 2023 13:25:24 +0200 Subject: [PATCH] use iterator as a default value in lambda --- ammico/multimodal_search.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/ammico/multimodal_search.py b/ammico/multimodal_search.py index 2bbf8aa..de7c208 100644 --- a/ammico/multimodal_search.py +++ b/ammico/multimodal_search.py @@ -17,7 +17,6 @@ from lavis.processors import load_processor class MultimodalSearch(AnalysisMethod): def __init__(self, subdict: dict) -> None: super().__init__(subdict) - # self.subdict.update(self.set_keys()) multimodal_device = torch.device("cuda" if torch.cuda.is_available() else "cpu") @@ -319,9 +318,12 @@ class MultimodalSearch(AnalysisMethod): ) similarity = features_image_stacked @ multi_features_stacked.t() - # similarity_soft_max = torch.nn.Softmax(dim=0)(similarity / 0.01) sorted_lists = [ - sorted(range(len(similarity)), key=lambda k: similarity[k, i], reverse=True) + sorted( + range(len(similarity)), + key=lambda k, value=i: similarity[k, value], + reverse=True, + ) for i in range(len(similarity[0])) ] places = [[item.index(i) for i in range(len(item))] for item in sorted_lists]