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.