search

HobbitLong / SupContrast

PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)
BSD 2-Clause "Simplified" License
3.08k stars 528 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 11 months ago

I have the same question.