locsec/AMMICO
1
Форкнуть 0
зеркало из https://github.com/ssciwr/AMMICO.git synced 2025-10-29 13:06:04 +02:00
Код Задачи Пакеты Проекты Релизы Вики Активность

uncommit part of the multimodal search test

Просмотр исходного кода
Этот коммит содержится в:
Petr Andriushchenko 2023-04-19 11:43:48 +02:00
родитель 903b1df644
Коммит fae982bc8b
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4C4A5DCF634115B6
1 изменённых файлов: 23 добавлений и 23 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

46
misinformation/test/test_multimodal_search.py
Просмотреть файл

@ -200,29 +200,29 @@ dict_image_gradcam_with_itm_for_blip = {
"pre_sorted",
),
[
# (
# device("cpu"),
# "blip2",
# pre_proc_pic_blip2_blip_albef,
# pre_proc_text_blip2_blip_albef,
# pre_extracted_feature_img_blip2,
# pre_extracted_feature_text_blip2,
# simularity_blip2,
# sorted_blip2,
# ),
# pytest.param(
# device("cuda"),
# "blip2",
# pre_proc_pic_blip2_blip_albef,
# pre_proc_text_blip2_blip_albef,
# pre_extracted_feature_img_blip2,
# pre_extracted_feature_text_blip2,
# simularity_blip2,
# sorted_blip2,
# marks=pytest.mark.skipif(
# gpu_is_not_available, reason="gpu_is_not_availible"
# ),
# ),
(
device("cpu"),
"blip2",
pre_proc_pic_blip2_blip_albef,
pre_proc_text_blip2_blip_albef,
pre_extracted_feature_img_blip2,
pre_extracted_feature_text_blip2,
simularity_blip2,
sorted_blip2,
),
pytest.param(
device("cuda"),
"blip2",
pre_proc_pic_blip2_blip_albef,
pre_proc_text_blip2_blip_albef,
pre_extracted_feature_img_blip2,
pre_extracted_feature_text_blip2,
simularity_blip2,
sorted_blip2,
marks=pytest.mark.skipif(
gpu_is_not_available, reason="gpu_is_not_availible"
),
),
(
device("cpu"),
"blip",