Chapter 3 - Making Predictions Method from Chapter 2 does not work about beginners-pytorch-deep-learning HOT 4 CLOSED

@MarcusFra : My mistake was to look into the old repository and mix the code up. This one here has the correct syntax.

from beginners-pytorch-deep-learning.

Hi @Mahosaurus!
Could you please elaborate a bit more what issue has arisen?
Did you need to replace the model in order to run the code?
Have you replaced the code line

img = torch.unsqueeze(img, 0)

in https://github.com/falloutdurham/beginners-pytorch-deep-learning/blob/master/chapter2/Chapter%202.ipynb?

If you still have the stacktrace I'd be thankful if you can share it.

from beginners-pytorch-deep-learning.

Ah, sorry my fault. You were referring to chapter 3. World you like to share your whole notebook with your solution?

from beginners-pytorch-deep-learning.

Hi @MarcusFra: Thanks for responding. I cant share the notebook as I have been using it for further experimentation.
The prediction part for chapter 3, I changed as follows in comp to chapter 2.

img = Image.open("file.jpg") 
img = img_transforms(img).unsqueeze(0)

prediction = F.softmax(alexcnnmodel(img))
prediction = prediction.argmax()

labels[prediction]

from beginners-pytorch-deep-learning.