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=tf.GPUOptions(allow_growth=True)))
+
         with self.graph.as_default():
 
             test_real = tf.placeholder(tf.float32, [1, None, None, 3], name='test')
@@ -20,6 +20,10 @@ class ImportGraph:
 
             self.saver = tf.train.Saver()
 
+            self.sess = tf.Session(graph=self.graph, config=tf.ConfigProto(allow_soft_placement=True,
+                                                                           gpu_options=tf.GPUOptions(
+                                                                               allow_growth=True)))
+
             ckpt = tf.train.get_checkpoint_state(checkpoint_dir)  # checkpoint file information
             if ckpt and ckpt.model_checkpoint_path:
                 ckpt_name = os.path.basename(ckpt.model_checkpoint_path)  # first line