pytorch / ignite

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
https://pytorch-ignite.ai
BSD 3-Clause "New" or "Revised" License
4.51k stars 611 forks source link

SimCLR example #1219

Open sdesrozis opened 4 years ago

sdesrozis commented 4 years ago

🚀 Feature

Idea is to provide new class of example for self supervised learning.

It could be an implementation of paper "A Simple Framework for Contrastive Learning of Visual Representations" https://arxiv.org/abs/2002.05709

Things to do are

vfdev-5 commented 4 years ago

This also may be a good FR for ignite.data module.

vfdev-5 commented 3 years ago

Maybe, it would be better to reimplement BYOL too : https://arxiv.org/abs/2006.07733

sdesrozis commented 3 years ago

Yes, it's not so far from SimCLR. I will do it in my research lab during next few months.