init

test1.py

@@ -11,7 +11,7 @@ os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 class ImportGraph:
     def __init__(self, checkpoint_dir):
         self.graph = tf.Graph()
-        self.sess = tf.Session(graph=self.graph, config=tf.ConfigProto(allow_soft_placement=True, gpu_options=gpu_options))
+        self.sess = tf.Session(graph=self.graph, config=tf.ConfigProto(allow_soft_placement=True, gpu_options=tf.GPUOptions(allow_growth=True)))
         with self.graph.as_default():
             test_real = tf.placeholder(tf.float32, [1, None, None, 3], name='test')
             with tf.variable_scope("generator", reuse=False):