Этот коммит содержится в:
Petr Andriushchenko 2023-04-27 14:30:55 +02:00
родитель df1ea34bbc
Коммит 4ba8b43659
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4C4A5DCF634115B6

Просмотреть файл

@ -318,7 +318,6 @@ class MultimodalSearch(AnalysisMethod):
)
similarity = features_image_stacked @ multi_features_stacked.t()
# similarity_soft_max = torch.nn.Softmax(dim=0)(similarity / 0.01)
sorted_lists = torch.argsort(similarity, dim=0, descending=True).T.tolist()
places = [[item.index(i) for i in range(len(item))] for item in sorted_lists]