Skip to content

Commit df38cc9

Browse files
committed
remove input shape
1 parent b6da6a2 commit df38cc9

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

��custom-train.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
class Model(keras.Model):
1111
def __init__(self):
1212
super(Model, self).__init__()
13-
self.l1 = keras.layers.Dense(10, activation=keras.activations.relu, input_shape=(1, ))
13+
self.l1 = keras.layers.Dense(10, activation=keras.activations.relu)
1414
self.l2 = keras.layers.Dense(1)
1515

1616
def call(self, x, training=None, mask=None):

‎simplified-training.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
test_x, test_y = data_x[900:], data_y[900:]
1010

1111
model = keras.models.Sequential([
12-
keras.layers.Dense(10, activation=keras.activations.relu, input_shape=(1, )),
12+
keras.layers.Dense(10, activation=keras.activations.relu),
1313
keras.layers.Dense(1),
1414
])
1515
model.compile(

0 commit comments

Comments
 (0)