I did targets = torch.unsqueeze(targets, 1) but now I get:
RuntimeError: output with shape [8, 1] doesn't match the broadcast shape [8, 8]. On my InceptionV3 I pass my num_classes = 1, so I guess it is returning one class for each batch (size = 8) when it should return some probabilistic number for each image of the batch. Can you help me please? I think I’m messing up hard