Последовательный объект не имеет атрибута «output_names» при сохранении модели keras

Question

Я использую Keras для загрузки данных mnist и трассировки, сохраняя данные keras.Я получаю следующую ошибку.полный код здесь ....

model = tf.keras.models.Sequential()
model.add(tf.keras.layers.Flatten())
model.add(tf.keras.layers.Dense(128, activation=tf.nn.relu))
model.add(tf.keras.layers.Dense(128, activation=tf.nn.relu))
model.add(tf.keras.layers.Dense(10, activation=tf.nn.softmax))

--------------------------------------------

 KERAS_MODEL_NAME = 'keras.hdf5'

tf.keras.models.save_model(model, KERAS_MODEL_NAME)
new_model=tf.keras.models.load_model(KERAS_MODEL_NAME)
predictions = new_model.predict([x_test])


getting following error 

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-1-be5d5fd55018> in <module>
     29 #tf.keras.models.save_model(model, KERAS_MODEL_NAME)
     30 tf.keras.models.save_model(model, KERAS_MODEL_NAME)
---> 31 new_model=tf.keras.models.load_model(KERAS_MODEL_NAME)
     32 predictions = new_model.predict([x_test])

AttributeError: 'Sequential' object has no attribute 'output_names'

Я также пытался все варианты, приведенные здесь , но не работал здесь ..