nb_cat = 3  # Number of images to concatenate
nb_classes=10
for epoch in range(2):
for i, data in enumerate(trainloader, 0):
inputs, labels = data
batch_size = inputs.size(0)
split_size = batch_size // nb_cat
inputs = torch.cat(inputs.split(split_size), 2)
labels = torch.stack(labels.split(nb_cat))
one_hot_labels = torch.zeros(inputs.size(0), nb_classes, dtype=torch.long).scatter_(1, labels, 1)
one_hot_labels = one_hot_labels.float()
this gives me this error:
RuntimeError: split_size can only be 0 if dimension size is 0, but got dimension size of 2