Update neural_style.py

neural_style.py

@@ -59,7 +59,7 @@ class TransferParams():
     style_blend_weights = None
     content_image = 'examples/inputs/tubingen.jpg'
     image_size = 300
-    gpu = "c"
+    gpu = 0# "c"
     content_weight = 5e0
     style_weight = 1e2
     normalize_weights = False
@@ -353,9 +353,9 @@ def setup_gpu():
             setup_cuda()
         dtype = torch.FloatTensor
 
-    #elif "c" not in str(params.gpu).lower():
-        #setup_cuda()
-        #dtype, backward_device = torch.cuda.FloatTensor, "cuda:" + str(params.gpu)
+    elif "c" not in str(params.gpu).lower():
+        setup_cuda()
+        dtype, backward_device = torch.cuda.FloatTensor, "cuda:" + str(params.gpu)
     else:
         setup_cpu()
         dtype, backward_device = torch.FloatTensor, "cpu"