Mike2004:

torch.argmax(output,dim=1)


It’s saying that this output torch.argmax(output,dim=1) is still long. can you print those tensors at the end you can actually see the type of the tensor or type(torch.argmax(output,dim=1))