From 04f9a4d02194970fde6a5a7e8018768e2df0504e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Maciej=20P=C4=99=C5=9Bko?= Date: Sat, 10 Nov 2018 13:11:39 +0100 Subject: [PATCH] Minor fix with sessions --- ComixGAN/model.py | 1 - neural_image_assessment/model.py | 1 - 2 files changed, 2 deletions(-) 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)