Fixed problem with GPU

Author: Hvass-Labs

vgg16.py

@@ -170,16 +170,20 @@ def create_feed_dict(self, image):
         # VGG16 model was built to take multiple images as input.
         image = np.expand_dims(image, axis=0)
 
-        # In the original code using this VGG16 model, the random values
-        # for the dropout are fixed to 1.0.
-        # Experiments suggest that it does not seem to matter for
-        # Style Transfer, but we do it anyway.
-        dropout_fix = 1.0
-
-        # Create feed-dict for inputting data to TensorFlow.
-        feed_dict = {self.tensor_name_input_image: image,
-                     self.tensor_name_dropout: [[dropout_fix]],
-                     self.tensor_name_dropout1: [[dropout_fix]]}
+        if False:
+            # In the original code using this VGG16 model, the random values
+            # for the dropout are fixed to 1.0.
+            # Experiments suggest that it does not seem to matter for
+            # Style Transfer, and this causes an error with a GPU.
+            dropout_fix = 1.0
+
+            # Create feed-dict for inputting data to TensorFlow.
+            feed_dict = {self.tensor_name_input_image: image,
+                         self.tensor_name_dropout: [[dropout_fix]],
+                         self.tensor_name_dropout1: [[dropout_fix]]}
+        else:
+            # Create feed-dict for inputting data to TensorFlow.
+            feed_dict = {self.tensor_name_input_image: image}
 
         return feed_dict