You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

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 .keras extension for the native Keras format (recommended) or a .h5 extension.

如果确实需要保存成你提到的“TensorFlow格式”(也就是SavedModel格式),有两种简单的解决办法:

  • 显式指定保存格式参数:
    model.save('my_model', save_format='saved_model')
    
  • 给保存路径加上.pb后缀,TensorFlow会自动识别为SavedModel格式:
    model.save('my_model.pb')
    

这样操作后,保存出来的就是你之前在官方文档和示例里看到的那种TensorFlow格式的模型啦。

备注:内容来源于stack exchange,提问作者Boppity Bop

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.04.14 08:25:29