From 6d22c2790292e1493a373076454917914fe3e642 Mon Sep 17 00:00:00 2001
From: Smiril <Smiril@users.noreply.github.com>
Date: Sun, 10 Jul 2022 21:58:42 +0200
Subject: [PATCH] Update m1_tf_test.py

---
 m1_tf_test.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/m1_tf_test.py b/m1_tf_test.py
index e74d577..56e739e 100644
--- a/m1_tf_test.py
+++ b/m1_tf_test.py
@@ -9,6 +9,8 @@
 import tensorflow as tf
 tf.__version__
 tf.config.list_physical_devices()
+logits = [[4.0, 2.0, 1.0], [0.0, 5.0, 1.0]]
+model = tf.keras.models.load_model('model')
 mnist = tf.keras.datasets.mnist
 (x_train, y_train), (x_test, y_test) = mnist.load_data()
 x_train, x_test = x_train / 255.0, x_test / 255.0
@@ -27,5 +29,5 @@ model.compile(optimizer='adam',
  metrics=['accuracy'])
 model.fit(x_train, y_train, epochs=10)
 model.build()
+model.save('model')
 model.summary()
-