prehistoric-systems/comixify
1
0
Fork 1
mirror of https://github.com/maciej3031/comixify.git synced 2026-03-11 08:54:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Minor fix with sessions

Browse source
This commit is contained in:
Maciej Pęśko 2018-11-10 13:11:39 +01:00
parent b2d059bc4d
commit 04f9a4d021
2 changed files with 0 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
ComixGAN/model.py
View file

@ -17,7 +17,6 @@ class ComixGAN:
config.gpu_options.allow_growth = True config.gpu_options.allow_growth = True
self.session = tf.Session(graph=self.graph, config=config) self.session = tf.Session(graph=self.graph, config=config)
with self.graph.as_default(): with self.graph.as_default():
self.session = tf.Session()
with self.session.as_default(): with self.session.as_default():
with tf.device('/device:GPU:0'): with tf.device('/device:GPU:0'):
self.model = load_model(settings.COMIX_GAN_MODEL_PATH, self.model = load_model(settings.COMIX_GAN_MODEL_PATH,

1
neural_image_assessment/model.py
View file

@ -20,7 +20,6 @@ class NeuralImageAssessment:
config.gpu_options.allow_growth = True config.gpu_options.allow_growth = True
self.session = tf.Session(graph=self.graph, config=config) self.session = tf.Session(graph=self.graph, config=config)
with self.graph.as_default(): with self.graph.as_default():
self.session = tf.Session()
with self.session.as_default(): with self.session.as_default():
self.model = load_model(settings.NIMA_MODEL_PATH) self.model = load_model(settings.NIMA_MODEL_PATH)