From 08550ac856cb606a46f3cd921537d99e50534fce Mon Sep 17 00:00:00 2001 From: iperov Date: Fri, 4 Jan 2019 18:28:07 +0400 Subject: [PATCH] fix H128 after refactoring --- models/Model_H128/Model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/Model_H128/Model.py b/models/Model_H128/Model.py index 48ffda3..857375e 100644 --- a/models/Model_H128/Model.py +++ b/models/Model_H128/Model.py @@ -174,7 +174,7 @@ class Model(ModelBase): y = upscale(256)(y) y = upscale(128)(y) else: - input_ = self.keras.layers.Input(shape=(16, 16, 256)) + input_ = Input(shape=(16, 16, 256)) x = input_ x = upscale(256)(x) x = upscale(128)(x)