From aabcb70c8c795c2067183d539586550ecf542e38 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Maciej=20P=C4=99=C5=9Bko?= Date: Sat, 10 Nov 2018 13:59:43 +0100 Subject: [PATCH] Add minor fix --- ComixGAN/model.py | 2 +- style_transfer/style_transfer.py | 3 --- 2 files changed, 1 insertion(+), 4 deletions(-) diff --git a/ComixGAN/model.py b/ComixGAN/model.py index c20d107..6d56278 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.7 + config.gpu_options.per_process_gpu_memory_fraction = 0.8 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 ace5826..2e68fb5 100644 --- a/style_transfer/style_transfer.py +++ b/style_transfer/style_transfer.py @@ -20,9 +20,6 @@ class StyleTransfer(): @classmethod @profile def get_stylized_frames(cls, frames, style_transfer_mode=0, gpu=settings.GPU): - print(frames[0].shape) - print(frames[0].max()) - print(frames[0].min()) if style_transfer_mode == 0: return cls._comix_gan_stylize(frames=frames) elif style_transfer_mode == 1: