fix H128 after refactoring
This commit is contained in:
@@ -174,7 +174,7 @@ class Model(ModelBase):
|
|||||||
y = upscale(256)(y)
|
y = upscale(256)(y)
|
||||||
y = upscale(128)(y)
|
y = upscale(128)(y)
|
||||||
else:
|
else:
|
||||||
input_ = self.keras.layers.Input(shape=(16, 16, 256))
|
input_ = Input(shape=(16, 16, 256))
|
||||||
x = input_
|
x = input_
|
||||||
x = upscale(256)(x)
|
x = upscale(256)(x)
|
||||||
x = upscale(128)(x)
|
x = upscale(128)(x)
|
||||||
|
|||||||
Reference in New Issue
Block a user