Fix some inference code to match the model with a log-softmax output

Author: mcleonard

intro-to-pytorch/Part 3 - Training Neural Networks (Exercises).ipynb

@@ -463,10 +463,10 @@
     "img = images[0].view(1, 784)\n",
     "# Turn off gradients to speed up this part\n",
     "with torch.no_grad():\n",
-    "    logits = model(img)\n",
+    "    logps = model(img)\n",
     "\n",
     "# Output of the network are logits, need to take softmax for probabilities\n",
-    "ps = F.softmax(logits, dim=1)\n",
+    "ps = torch.exp(logps)\n",
     "helper.view_classify(img.view(1, 28, 28), ps)"
    ]
   },