Keras load_model将永远在层名更新之后使用。

Question

我有一个定制的分类模型。由于某种原因，我需要更改上一次卷积层的名称。模型在更新前的7到9秒内加载，但是在名称更改之后，它将花费永远的时间。我找不到任何理由。首先，我使用tf.keras.callbacks.ModelCheckpoint()保存了模型。

我是这样更新的：

best_acc_model = tf.keras.models.load_model("path/to/model")
best_acc_model.get_layer("conv_name")._name = "new_conv_name"
best_acc_model.save("save/path")

我就是这样加载更新的模型的：

model = tf.keras.models.load_model("save/path", compile=False)

没有什么奇怪的，但有些如何永远加载和不引发任何例外。你有什么推荐吗？

Answer 1

解决办法是；

best_acc_model = tf.keras.models.load_model("path/to/model")
best_acc_model.get_layer("conv_name")._name = "new_conv_name"
best_acc_model = tf.keras.Model(inputs=best_acc_model.input, outputs=best_acc_model.output)
best_acc_model.save("save/path")

来源：链接到解决方案