在tensorflow1.1.4版本中出现cannot import name 'export_saved_model' from 'tensorflow.python.keras.saving.saved_model'的报错,该怎么办
时间: 2024-02-16 21:04:19 浏览: 229
MultiGet-1.1.4.rar_MultiGet-1.1_MultiGet-1.1.4.tar._VS C++多线程_mu
在TensorFlow 1.1.4版本中,`export_saved_model`函数是没有实现的,因此无法从`tensorflow.python.keras.saving.saved_model`中导入。
如果您需要导出SavedModel,可以使用下面的代码:
```python
import tensorflow as tf
# Build and train your model
# Export the model as a SavedModel
with tf.Session(graph=tf.Graph()) as sess:
# Build the model graph
# ...
# Train the model
# ...
# Export the model as a SavedModel
builder = tf.saved_model.builder.SavedModelBuilder('/path/to/export/dir')
builder.add_meta_graph_and_variables(sess, [tf.saved_model.tag_constants.SERVING], signature_def_map={
'predict': predict_signature_def,
'train': train_signature_def,
# Add additional signature defs as needed
})
builder.save()
```
请注意,此代码是基于TensorFlow 1.1.4版本,如果您使用的是其他版本,则可能需要进行适当的修改。
阅读全文