This PR fixes error raised during element-wise variable division. (py…

…torch#182) - Fixes pytorch#181
ownway22 · Jul 20, 2017 · 08be28e · 08be28e
1 parent 2d0f1c4
commit 08be28e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fast_neural_style/neural_style/utils.py b/fast_neural_style/neural_style/utils.py
@@ -39,5 +39,5 @@ def normalize_batch(batch):
     std[:, 2, :, :] = 0.225
     batch = torch.div(batch, 255.0)
     batch -= Variable(mean)
-    batch /= Variable(std)
+    batch = batch / Variable(std)
     return batch