That sounds right.


If you want to calculate the “test loss” (and have targets during inference), you can use the criterion to calculate it. Otherwise, if you just want to predictions, it’s not necesary.


The predictions won’t change and you can get the predicted class index via torch.argmax(output), where output can be the logits or the probabilities.