From 31e0e8328ab383a9f30c62927a8380a3dc7dff3a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Maciej=20P=C4=99=C5=9Bko?= Date: Sat, 10 Nov 2018 21:59:33 +0100 Subject: [PATCH] Change parameters --- ComixGAN/model.py | 2 +- neural_image_assessment/model.py | 2 +- style_transfer/style_transfer.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/ComixGAN/model.py b/ComixGAN/model.py index 6d56278..ec3439f 100644 --- a/ComixGAN/model.py +++ b/ComixGAN/model.py @@ -13,7 +13,7 @@ class ComixGAN: raise FileNotFoundError(errno.ENOENT, os.strerror(errno.ENOENT), settings.COMIX_GAN_MODEL_PATH) self.graph = tf.Graph() config = tf.ConfigProto() - config.gpu_options.per_process_gpu_memory_fraction = 0.8 + config.gpu_options.per_process_gpu_memory_fraction = 0.6 config.gpu_options.allow_growth = True self.session = tf.Session(graph=self.graph, config=config) with self.graph.as_default(): diff --git a/neural_image_assessment/model.py b/neural_image_assessment/model.py index feead33..18731cd 100644 --- a/neural_image_assessment/model.py +++ b/neural_image_assessment/model.py @@ -16,7 +16,7 @@ class NeuralImageAssessment: raise FileNotFoundError(errno.ENOENT, os.strerror(errno.ENOENT), settings.NIMA_MODEL_PATH) self.graph = tf.Graph() config = tf.ConfigProto() - config.gpu_options.per_process_gpu_memory_fraction = 0.8 + config.gpu_options.per_process_gpu_memory_fraction = 0.6 config.gpu_options.allow_growth = True self.session = tf.Session(graph=self.graph, config=config) with self.graph.as_default(): diff --git a/style_transfer/style_transfer.py b/style_transfer/style_transfer.py index 42ef6fc..f45c896 100644 --- a/style_transfer/style_transfer.py +++ b/style_transfer/style_transfer.py @@ -51,7 +51,7 @@ class StyleTransfer(): with comixGAN.graph.as_default(): with comixGAN.session.as_default(): - batch_size = 5 + batch_size = 4 stylized_imgs = [] for i in range(0, len(frames), batch_size): batch_of_frames = ((np.stack(frames[i:i + batch_size]) / 255) * 2) - 1