From 9471186c0a9f3e3042760e7309328a9f86ecc545 Mon Sep 17 00:00:00 2001 From: Smiril Date: Mon, 11 Jul 2022 00:10:23 +0200 Subject: [PATCH] Update m1_tf_test.py --- m1_tf_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/m1_tf_test.py b/m1_tf_test.py index 72acc0b..60814dd 100644 --- a/m1_tf_test.py +++ b/m1_tf_test.py @@ -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'),