TensorFlow 2.18版本中如何以TensorFlow格式保存检查点?
TensorFlow 2.18版本中如何以TensorFlow格式保存检查点?
我完全懂你的困惑!从TensorFlow 2.10版本开始,Keras的默认保存格式切换成了原生的.keras格式(官方也更推荐使用这个格式),所以直接执行model.save('something')就会触发你遇到的报错:
Invalid filepath extension for saving. Please add either a
.kerasextension for the native Keras format (recommended) or a.h5extension.
如果确实需要保存成你提到的“TensorFlow格式”(也就是SavedModel格式),有两种简单的解决办法:
- 显式指定保存格式参数:
model.save('my_model', save_format='saved_model') - 给保存路径加上
.pb后缀,TensorFlow会自动识别为SavedModel格式:model.save('my_model.pb')
这样操作后,保存出来的就是你之前在官方文档和示例里看到的那种TensorFlow格式的模型啦。
备注:内容来源于stack exchange,提问作者Boppity Bop
相关产品推荐
相关产品推荐

