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.

---

Проект выполнили: Алексей Подчезерцев, Мария Самоделкина