hse_DCL_positive
Анализ смещения распределений в контрастивном обучении
- datasets.py - модифицированные версии датасетов STL10 и CIFAR10
- loss.py - реализации функций потерь Contrastive, DebiasedNeg и DebiasedPos
- model.py - модифицированная версия модели ResNet
- utils.py - утилиты, аугментации
- main.py - основная точка входа, содержит скрипты для обучения модели
- main.ipynb - пример обучения модели в среде Jupyter Lab
- plot_experiments.ipynb - ноутбук с визуализацией результатов экспериментов
Acknowledgements
Part of this code is inspired by chingyaoc/DCL.
Проект выполнили: Алексей Подчезерцев, Мария Самоделкина