From fb24d4b1563e58245b9793df976c7d2c37e68e04 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Maciej=20P=C4=99=C5=9Bko?= <mpesko3031@gmail.com>
Date: Sat, 10 Nov 2018 12:52:13 +0100
Subject: [PATCH] Add minor fix with GPU name

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

diff --git a/ComixGAN/model.py b/ComixGAN/model.py
index d0cba49..96f4473 100644
--- a/ComixGAN/model.py
+++ b/ComixGAN/model.py
@@ -17,6 +17,6 @@ class ComixGAN:
         config.gpu_options.allow_growth = True
         self.session = tf.Session(graph=self.graph, config=config)
         with self.graph.as_default():
-            with tf.device('/GPU:0'):
+            with tf.device('/device:GPU:0'):
                 self.model = load_model(settings.COMIX_GAN_MODEL_PATH,
                                         custom_objects={'InstanceNormalization': InstanceNormalization})