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