Hi again,
One more quick question.
So I am able to classify the image using the code you listed above, but my output is the following:
Variable containing:
0
[torch.LongTensor of size 1]

How do I extract the value of this variable and convert it to, say, an int?
I want to use that int to print out the string description of the category it belongs to.
As I understand it, the value in this variable is the index of the class (folder) that it predicts the image belongs to?
Many thanks