Fix prediction error in transfer learning. (#119)

cxxgtxy · chsasank · commit b0cb64a58541 · 2017-07-24T14:41:01.000+05:30
* Fix prediction error.

* Fix spelling error in neural_style transfer.
diff --git a/advanced_source/neural_style_tutorial.py b/advanced_source/neural_style_tutorial.py
@@ -179,7 +179,7 @@
 # .. Note::
 #     Here are links to download the images required to run the tutorial:
 #     `picasso.jpg <http://pytorch.org/tutorials/_static/img/neural-style/picasso.jpg>`__ and
-#     `dancing.jpg <http://pytorch.org/tutorials/_static/img/neural-style/picasso.jpg>`__.
+#     `dancing.jpg <http://pytorch.org/tutorials/_static/img/neural-style/dancing.jpg>`__.
 #     Download these two images and add them to a directory
 #     with name ``images``
 
@@ -269,8 +269,8 @@ def imshow(tensor, title=None):
 #
 # Finally, we define a fake ``backward`` method, that just call the
 # backward method of ``nn.MSELoss`` in order to reconstruct the gradient.
-# This method returns the computed loss: this will be usefull when running
-# the gradien descent in order to display the evolution of style and
+# This method returns the computed loss: this will be useful when running
+# the gradient descent in order to display the evolution of style and
 # content losses.
 #
 
diff --git a/beginner_source/transfer_learning_tutorial.py b/beginner_source/transfer_learning_tutorial.py
@@ -253,7 +253,7 @@ def visualize_model(model, num_images=6):
             images_so_far += 1
             ax = plt.subplot(num_images//2, 2, images_so_far)
             ax.axis('off')
-            ax.set_title('predicted: {}'.format(dset_classes[labels.data[j]]))
+            ax.set_title('predicted: {}'.format(dset_classes[preds[j][0]]))
             imshow(inputs.cpu().data[j])
 
             if images_so_far == num_images:
