在tensorflow.keras.experimental模型中,可以通过tf.saved_model.save
函数来导出模型。具体来说,可以通过以下步骤找到export_saved_model属性:
import tensorflow as tf
from tensorflow.keras import layers
model = tf.keras.Sequential([
layers.Dense(64, activation='relu', input_shape=(784,)),
layers.Dense(10, activation='softmax')
])
model.compile(optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])
model.fit(x_train, y_train, epochs=10)
export_path = '/path/to/exported/model'
tf.saved_model.save(model, export_path)
在上述代码中,export_saved_model
属性即为tf.saved_model.save
函数,它用于将模型保存到指定路径。导出的模型可以用于后续的推理或部署。
腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云