Update Neural Style Transfer to print weighted loss (#225)

ssnl · chsasank · commit 2543feecf0ac · 2018-04-25T15:12:34.000+05:30
* Update Neural Style Transfer to print weighted loss

* fix a typo too
diff --git a/advanced_source/neural_style_tutorial.py b/advanced_source/neural_style_tutorial.py
@@ -154,7 +154,7 @@
 # VGG). For this, we have ``torch.cuda.is_available()`` that returns
 # ``True`` if you computer has an available GPU. Then, we can set the
 # ``torch.device`` that will be used in this script. Then, we will use
-# the method ``.to(device)`` that a tensor or a module to the desired
+# the method ``.to(device)`` that moves a tensor or a module to the desired
 # device. When we want to move back this tensor or module to the
 # CPU (e.g. to use numpy), we can use the ``.cpu()`` method.
 
@@ -561,7 +561,11 @@ def closure():
             for cl in content_losses:
                 content_score += cl.loss
 
-            (style_score * style_weight + content_score * content_weight).backward()
+            style_score *= style_weight
+            content_score *= content_weight
+
+            loss = style_score + content_score
+            loss.backward()
 
             run[0] += 1
             if run[0] % 50 == 0:
