Dear all:
when I read the code of pytorch transform_learnning tuitorials, I see the code below:
outputs = model(inputs)
_, preds = torch.max(outputs, 1)
loss = criterion(outputs, labels)
it seems that  mistake in the documentation of the loss funciton.
i think that the loss should be like as:
loss = criterion(preds, labels)
Thanks .