How to Classify Single Image using Loaded Net

16

  1. one of them (model or input) is in cpu and the other is in GPU.
    Upload them to GPU using model = model.cuda(), image = image.cuda() commands.

  2. The output that we get from pretrained models are unnormalized linear layer outputs.
    you can use F.softmax() to convert them to probabilities.