From 99c5f2ad6755036e174daebfe183e53c16268747 Mon Sep 17 00:00:00 2001 From: Doojin Park Date: Wed, 27 Nov 2024 04:33:43 +0000 Subject: [PATCH] rollback attribute changes which is not compatible with TF 1.15 --- tfx/examples/mnist/mnist_utils_native_keras_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tfx/examples/mnist/mnist_utils_native_keras_base.py b/tfx/examples/mnist/mnist_utils_native_keras_base.py index eae1ebc7d6..965988d3a6 100644 --- a/tfx/examples/mnist/mnist_utils_native_keras_base.py +++ b/tfx/examples/mnist/mnist_utils_native_keras_base.py @@ -69,7 +69,7 @@ def build_keras_model() -> tf.keras.Model: model = tf.keras.Sequential() model.add( tf.keras.layers.InputLayer( - shape=(784,), name=transformed_name(IMAGE_KEY))) + input_shape=(784,), name=transformed_name(IMAGE_KEY))) model.add(tf.keras.layers.Dense(64, activation='relu')) model.add(tf.keras.layers.Dropout(0.2)) model.add(tf.keras.layers.Dense(64, activation='relu'))