fix H128 after refactoring

2019-01-04 18:28:07 +04:00
parent 1398072427
commit 08550ac856
1 changed files with 1 additions and 1 deletions
--- 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)