在哪里可以找到tensorflow.keras.experimental模型中的export_saved_model属性？

Question

我已经运行了该模型。现在训练后导出模型时出现错误。

这是代码：

SHOE_SAVED_MODEL = "saved_models/shoe"
tf.keras.experimental.export_saved_model(model, SHOE_SAVED_MODEL)
shoe_model = tf.keras.experimental.load_from_saved_model(SHOE_SAVED_MODEL,
custom_objects={'KerasLayer':hub.KerasLayer})

代码中的错误：

AttributeError                            Traceback (most recent call last)
<ipython-input-24-59c886611cb2> in <module>()
      1 SHOE_SAVED_MODEL = "saved_models/shoe"
----> 2 tf.keras.experimental.export_saved_model(model, SHOE_SAVED_MODEL)
      3 shoe_model = tf.keras.experimental.load_from_saved_model(SHOE_SAVED_MODEL,
      4 custom_objects={'KerasLayer':hub.KerasLayer})

AttributeError: module 'tensorflow.keras.experimental' has no attribute 'export_saved_model'

Answer 1

截至目前，该功能tensorflow.keras.experimental不再起作用。要保存模型，请使用该model.save('path')功能

saved_model_path = "./my_model.h5". # or you can simply use 'my_mode.h5'
model.save(saved_model_path) #save your model 

# to reload your model 
model = keras.models.load_model(saved_model_path)