From 8e8e0e08a20ef8f42a4c96944c9003a396064cca Mon Sep 17 00:00:00 2001 From: Smiril Date: Sun, 10 Jul 2022 22:53:56 +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 b536111..41a7382 100644 --- a/m1_tf_test.py +++ b/m1_tf_test.py @@ -33,7 +33,7 @@ model.compile(optimizer='adam', loss=loss_fn, metrics=['accuracy']) model.fit(x_test, y_test, epochs=10) -outputs = tf.keras.layers.Dense(4, activation="softmax", name="predictions")(x_test) +outputs = tf.keras.layers.Dense(4, activation='softmax', name='predictions')((x_test,y_test)) model = tf.keras.Model(inputs=inputs, outputs=outputs) model.build() model.save('model')