targets.append(torch.FloatTensor(sample[1])) RuntimeError: given sequence has an invalid size of dimension 2: 0