target size

[xysong1201]

Hi, ptrblck, I don't know why your target tensor the final dimension is a tuple. could you explain y = torch.randint(0, nb_classes, (1, 96, 96))

[ptrblck]

torch.randint expects arguments as low, high, size, out, .... Since I wanted to sample random integers in [0, nb_classes-1] in the shape [1, 96, 96], I had to pass it this way. Otherwise, the 96 might be understood as the out argument.