SupContrast

a pytorch implement of Supervised Contrastive Learning with memory bank(queue)

Original paper set batch_size = 6144.........

So, I use the queue memory bank proposed by momentum contrastive learning while use supervised method, so this code is a combination of these 2 paper.