Update m1_tf_test.py

2022-07-11 00:10:23 +02:00
parent 3bc7c48e02
commit 9471186c0a
1 changed files with 1 additions and 1 deletions
@@ -49,7 +49,7 @@ mnist = tf.keras.datasets.mnist
 (x_train, y_train), (x_test, y_test) = mnist.load_data()
 x_train, x_test = x_train / randrange(255), x_test / randrange(255)
 model = tf.keras.models.Sequential([
- tf.keras.layers.Flatten(input_shape=(randrange(255), randrange(255))),
+ tf.keras.layers.Flatten(input_shape=(28, 28)),
 tf.keras.layers.Dense(128,activation='selu',name='layer1'),
 tf.keras.layers.Dropout(0.2),
 tf.keras.layers.Dense(64,activation='relu',name='layer2'),