diff --git a/ComixGAN/model.py b/ComixGAN/model.py index 85d9c75..c20d107 100644 --- a/ComixGAN/model.py +++ b/ComixGAN/model.py @@ -17,7 +17,6 @@ class ComixGAN: config.gpu_options.allow_growth = True self.session = tf.Session(graph=self.graph, config=config) with self.graph.as_default(): - self.session = tf.Session() with self.session.as_default(): with tf.device('/device:GPU:0'): self.model = load_model(settings.COMIX_GAN_MODEL_PATH, diff --git a/neural_image_assessment/model.py b/neural_image_assessment/model.py index 0f5f48c..44e1b0f 100644 --- a/neural_image_assessment/model.py +++ b/neural_image_assessment/model.py @@ -20,7 +20,6 @@ class NeuralImageAssessment: config.gpu_options.allow_growth = True self.session = tf.Session(graph=self.graph, config=config) with self.graph.as_default(): - self.session = tf.Session() with self.session.as_default(): self.model = load_model(settings.NIMA_MODEL_PATH)