Should assert n_views == 2?

[LinglanZhao]

Thanks for your excellent implementation! I'd like to confirm that N_VIEW == 2 as in the paper and the default args in the code. If N_VIEW > 2, with logits.shape = (N_VIEW x N, N_VIEW x N - 1)

SimCLR/simclr.py

Line 51 in 1848fc9

logits = torch.cat([positives, negatives], dim=1)

, N_VIEW x N - 1 contains at least one more positive pairs (except the one indexed with 0) which will be treated as negative pairs. @sthalles @alessiamarcolini @butyuhao

[here101]

Have you solved the problem?
I meet the same question.
Thanks.