Update neural_style.py

neural_style.py

@@ -70,7 +70,7 @@ class TransferParams():
     optimizer = 'lbfgs'
     learning_rate = 1e0
     lbfgs_num_correction = 100
-    print_iter = 50
+    print_iter = 1000
     save_iter = 1000
     output_image = 'out.png'
     log_level = 10
@@ -91,7 +91,7 @@ def main():
 
 def transfer(params):
     dtype, multidevice, backward_device = setup_gpu()
-    print(dtype)
+ 
 
     cnn, layerList = loadCaffemodel(params.model_file, params.pooling, params.gpu, params.disable_check)
 
@@ -156,13 +156,13 @@ def transfer(params):
                 net.add_module(str(len(net)), layer)
 
                 if layerList['C'][c] in content_layers:
-                    print("Setting up content layer " + str(i) + ": " + str(layerList['C'][c]))
+                    #print("Setting up content layer " + str(i) + ": " + str(layerList['C'][c]))
                     loss_module = ContentLoss(params.content_weight)
                     net.add_module(str(len(net)), loss_module)
                     content_losses.append(loss_module)
 
                 if layerList['C'][c] in style_layers:
-                    print("Setting up style layer " + str(i) + ": " + str(layerList['C'][c]))
+                    #print("Setting up style layer " + str(i) + ": " + str(layerList['C'][c]))
                     loss_module = StyleLoss(params.style_weight)
                     net.add_module(str(len(net)), loss_module)
                     style_losses.append(loss_module)
@@ -172,14 +172,14 @@ def transfer(params):
                 net.add_module(str(len(net)), layer)
 
                 if layerList['R'][r] in content_layers:
-                    print("Setting up content layer " + str(i) + ": " + str(layerList['R'][r]))
+                    #print("Setting up content layer " + str(i) + ": " + str(layerList['R'][r]))
                     loss_module = ContentLoss(params.content_weight)
                     net.add_module(str(len(net)), loss_module)
                     content_losses.append(loss_module)
                     next_content_idx += 1
 
                 if layerList['R'][r] in style_layers:
-                    print("Setting up style layer " + str(i) + ": " + str(layerList['R'][r]))
+                    #print("Setting up style layer " + str(i) + ": " + str(layerList['R'][r]))
                     loss_module = StyleLoss(params.style_weight)
                     net.add_module(str(len(net)), loss_module)
                     style_losses.append(loss_module)
@@ -195,7 +195,7 @@ def transfer(params):
     # Capture content targets
     for i in content_losses:
         i.mode = 'capture'
-    print("Capturing content targets")
+    #print("Capturing content targets")
     print_torch(net, multidevice)
     net(content_image)
 
@@ -204,7 +204,7 @@ def transfer(params):
         i.mode = 'None'
 
     for i, image in enumerate(style_images_caffe):
-        print("Capturing style target " + str(i+1))
+        #print("Capturing style target " + str(i+1))
         for j in style_losses:
             j.mode = 'capture'
             j.blend_weight = style_blend_weights[i]