From 3bc7c48e0268062958af07211e98008e0af33e1f Mon Sep 17 00:00:00 2001 From: Smiril Date: Mon, 11 Jul 2022 00:04:50 +0200 Subject: [PATCH] Update m1_tf_test.py --- m1_tf_test.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/m1_tf_test.py b/m1_tf_test.py index ac193e1..72acc0b 100644 --- a/m1_tf_test.py +++ b/m1_tf_test.py @@ -43,8 +43,7 @@ def xatoi(Str): return base * sign -user_input = sys.argv[1] -inputs = tf.keras.Input(shape=(xatoi(user_input),), name="digits") +inputs = tf.keras.Input(shape=(xatoi(sys.argv[1]),), name="digits") model = tf.keras.models.load_model('model') mnist = tf.keras.datasets.mnist (x_train, y_train), (x_test, y_test) = mnist.load_data()