From 6432eaafcfa257a70a4fb507e445113e18c511bc Mon Sep 17 00:00:00 2001 From: Smiril Date: Mon, 11 Jul 2022 10:50:33 +0200 Subject: [PATCH] Update m1_tf_test.py --- m1_tf_test.py | 1 - 1 file changed, 1 deletion(-) diff --git a/m1_tf_test.py b/m1_tf_test.py index 436db2f..aab7601 100644 --- a/m1_tf_test.py +++ b/m1_tf_test.py @@ -93,7 +93,6 @@ model.compile(optimizer='adam', loss=loss_fn, metrics=['accuracy']) these = model.fit(x_train, y_train, epochs=10).history that = ComputeSum(len(these)) those = that(these) -outputs = tf.keras.layers.Dense(4, activation='softmax', name='predictions')(those) model = ComputeSumModel((tf.shape(these)[1],)) model = tf.keras.Model(inputs=inputs, outputs=outputs) model.build()