From acb0cccfb1d67112f2e296a25dd7bb83793b891a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Maciej=20P=C4=99=C5=9Bko?= <mpesko3031@gmail.com>
Date: Sun, 4 Nov 2018 20:10:49 +0100
Subject: [PATCH] Add tmp solution for GPU ram problem

---
 style_transfer/style_transfer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/style_transfer/style_transfer.py b/style_transfer/style_transfer.py
index b5fa734..7d3a3eb 100644
--- a/style_transfer/style_transfer.py
+++ b/style_transfer/style_transfer.py
@@ -57,7 +57,7 @@ class StyleTransfer():
         comixGAN_model = load_model(settings.COMIX_GAN_MODEL_PATH,
                                     custom_objects={'InstanceNormalization': InstanceNormalization})
         frames = cls._resize_images(frames, size=450)
-        batch_size = 5
+        batch_size = 2
         stylized_imgs = []
         for i in range(0, len(frames), batch_size):
             batch_of_frames = np.stack(frames[i:i + batch_size]) / 255