search

HobbitLong / SupContrast

PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)
BSD 2-Clause "Simplified" License
3.12k stars 537 forks source link

f1, f2 = torch.split(features, [bsz, bsz], dim=0) #112

Open onlinehuazai opened 2 years ago

onlinehuazai commented 2 years ago

f1, f2 = torch.split(features, [bsz, bsz], dim=0)

RuntimeError: start (4) + length (4) exceeds dimension size (4).

1imy commented 1 year ago

I have the same question.