Tensorflow2.x 模型保存和加载（ValueError: No model found in config file.）

最新推荐文章于 2024-03-24 16:40:13 发布

Hubinbin2018

最新推荐文章于 2024-03-24 16:40:13 发布

阅读量7.1k

点赞数 3

分类专栏： tensorflow python

本文链接：https://blog.youkuaiyun.com/benzp/article/details/106634535

版权

Tensorflow2.x 中，模型保存有两种格式：SavedModel 和 HDF5。SavedModel 支持自定义对象，而 HDF5 需要实现 `get_config` 和 `from_config` 方法来保存和加载包含自定义对象的模型。若使用 HDF5 格式，需将自定义对象传入 `custom_objects` 参数以避免 `ValueError: No model found in config file.` 错误。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Tensorflow2.x 模型保存和加载

Tensorflow官网教程给出了两种保存整个模型的方法：

SavedModel format
HDF5 format

在自定义模型时遇到了一个问题：

Traceback (most recent call last):
  File "/data/projects/sysml/resNetAll/main.py", line 100, in <module>
    'learning_rate':learning_rate,'decay_rate':decay_rate,
  File "/usr/local/lib/python3.6/dist-packages/t