From 8c51888f712d17ed2252fae14de0e6051f975a90 Mon Sep 17 00:00:00 2001
From: nl <3210346136@qq.com>
Date: Wed, 26 Sep 2018 19:45:53 +0800
Subject: [PATCH] fix init bug for multi gpu model

---
 general_function/muti_gpu.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/general_function/muti_gpu.py b/general_function/muti_gpu.py
index 621d1f2..6b13577 100644
--- a/general_function/muti_gpu.py
+++ b/general_function/muti_gpu.py
@@ -24,6 +24,7 @@ class ParallelModel(keras.models.Model):
         keras_model: The Keras model to parallelize
         gpu_count: Number of GPUs. Must be > 1
         """
+		super(ParallelModel, self).__init__() # Thanks to @greatken999 for fixing bugs
         self.inner_model = keras_model
         self.gpu_count = gpu_count
         merged_outputs = self.make_parallel()