python - UserWarning : No training configuration found in save file: the model was *not* compiled. 手动编译

Question

在训练过程之后，我想通过加载创建的 model.h5 并执行评估过程来检查准确性。但是，我收到以下警告:

/usr/local/lib/python3.5/dist-packages/keras/engine/saving.py:269: UserWarning: No training configuration found in save file: the model was not compiled. Compile it manually. warnings.warn('No training configuration found in save file:

这个dist-packages/keras/engine/saving.py 文件

所以加载创建模型的问题->这行代码

train_model = load_model('model.h5')

问题表明模型没有编译，但是我编译了。

optimizer = Adam(lr=lr, clipnorm=0.001)
train_model.compile(loss=dummy_loss, optimizer=optimizer)

我不明白我做错了什么。 . .请帮我!紧急求救:-(

Answer 1

简介

我想补充 olejorgenb 的回答 - 对于特定场景，您不想训练模型，只需使用它(例如在生产中)。

“编译”的意思是“准备训练”，主要包括设置优化器。也可以之前保存过，然后载入保存的模型就可以继续“同样”的训练了。

修复

但是，场景如何 - 我只想运行模型？好吧，像这样使用 compile=False 参数到 load_model:

trained_model = load_model('model.h5', compile=False)

如果不首先使用 trained_model.compile(...)，您将无法.fit() 这个模型，但最重要的是 - 警告将会消失离开。

杂项笔记

顺便说一句，在我的 Keras 版本中，参数 include_optimizer 的默认值为 True。这也适用于 Checkpoint 等训练回调。这意味着，在加载由 Keras 保存的模型时，您通常可以指望包含优化器(情况除外:请参阅 Hull Gasper 的回答)。

但是，当您有一个未经 Keras 训练的模型时(例如，当转换一个由 Darknet 训练的模型时)，该模型将以未编译的方式保存。这会产生警告，您可以按照上述方式摆脱它。