Based on the demo code it looks like one demo is using a pretrained torchvision.models.resnet18.
This model outputs logits for each class, such that the highest logit would correspond to the predicted class. You could apply a softmax, which is not necessary to get the prediction, but will give you probabilities, which might be easier to interpret.


If I’m right in point 1, you should see positive and negative numbers as the logit output.


You could add a softmax layer to your model.