Update neural_style.py

neural_style.py

@@ -249,7 +249,7 @@ def transfer(params):
                 print("  Style " + str(i+1) + " loss: " + str(loss_module.loss.item()))
             print("  Total loss: " + str(loss.item()))
 
-    final_image = ''
+    #final_image = ''
     def maybe_save(t):
         should_save = params.save_iter > 950 and t % params.save_iter == 0
         should_save = should_save or t == params.num_iterations
@@ -297,7 +297,7 @@ def transfer(params):
         maybe_print(num_calls[0], loss)
 
         return loss
-    print('the final image is', final_image)
+    ##print('the final image is', final_image)
     optimizer, loopVal = setup_optimizer(img)
     while num_calls[0] <= loopVal:
          optimizer.step(feval)
@@ -411,7 +411,7 @@ def print_torch(net, multidevice):
     simplelist = ""
     for i, layer in enumerate(net, 1):
         simplelist = simplelist + "(" + str(i) + ") -> "
-    print("nn.Sequential ( \n  [input -> " + simplelist + "output]")
+    #print("nn.Sequential ( \n  [input -> " + simplelist + "output]")
 
     def strip(x):
         return str(x).replace(", ",',').replace("(",'').replace(")",'') + ", "