B2BMGMT_Cartoonize

Running

imseldrith commited on Apr 5

Commit

42bf31d

•

1 Parent(s): ca598b4

Update white_box_cartoonizer/cartoonize.py

Files changed (1) hide show

white_box_cartoonizer/cartoonize.py CHANGED Viewed

@@ -64,7 +64,7 @@ class WB_Cartoonize:
             gpu_options = None
             device_count = {'GPU':0}
-        config = tf.ConfigProto(gpu_options=None, device_count={'GPU':0})
         self.sess = tf.Session(config=config)

             gpu_options = None
             device_count = {'GPU':0}
+        config = tf.ConfigProto(gpu_options=gpu_options, device_count=device_count)
         self.sess = tf.Session(config=config)