removed unnecessary kwarg
This commit is contained in:
parent
d11333981f
commit
0f811bdc56
@ -58,7 +58,7 @@ def keras_model(input_size):
|
|||||||
|
|
||||||
inputs = tf.keras.Input(shape=input_size)
|
inputs = tf.keras.Input(shape=input_size)
|
||||||
conv = tf.keras.layers.Conv2D(3, 3)
|
conv = tf.keras.layers.Conv2D(3, 3)
|
||||||
dense = tf.keras.layers.Dense(10, activation="relu")
|
dense = tf.keras.layers.Dense(10)
|
||||||
|
|
||||||
outputs = tf.keras.layers.Dense(10)(dense(conv(inputs)))
|
outputs = tf.keras.layers.Dense(10)(dense(conv(inputs)))
|
||||||
model = tf.keras.Model(inputs=inputs, outputs=outputs)
|
model = tf.keras.Model(inputs=inputs, outputs=outputs)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user