From 788d1e57a2d24be8f8daa4a342bae36e81ddcae5 Mon Sep 17 00:00:00 2001 From: Smiril Date: Mon, 11 Jul 2022 10:42:04 +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 5dbb158..d18b0c2 100644 --- a/m1_tf_test.py +++ b/m1_tf_test.py @@ -89,7 +89,7 @@ model = tf.keras.models.Sequential([ tf.keras.layers.Dropout(0.2) ]) loss_fn = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True) -model.add(LSTM(28, input_shape=(10, 1))) +model.add(LSTM(1, input_shape=(10, 1))) model.compile(optimizer='adam', loss=loss_fn, metrics=['accuracy']) these = model.fit(x_train, y_train, epochs=10).history that = ComputeSum(len(these))