Update m1_tf_test.py

2022-07-11 00:04:50 +02:00
parent c03cf92a4e
commit 3bc7c48e02
1 changed files with 1 additions and 2 deletions
@@ -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()