如何使用自定义优化器加载 keras 保存的模型

Question

我已经使用自定义优化器编译和训练了 keras 模型。我保存了模型，但是当我尝试加载模型时，它抛出一个错误，指出ValueError: Unknown optimizer: MyOptimizer. 我试图将 MyOptimizer 作为自定义对象传递，例如 :models.load_model('myModel.h5', custom_objects={'optimizer':MyOptimizer})并且它仍然抛出错误。如何使用自定义对象将模型加载到 keras 模型？

Answer 1

我遇到了同样的问题:)

我通过加载模型来使它工作models.load_model('myModel.h5', compile=False)。

从 keras源代码：

如果发现优化器作为保存模型的一部分，则模型已经编译。否则，模型未编译并显示警告。当compile设置为 False 时，编译将被省略，没有任何警告。

加载未编译的模型后，我可以使用我的自定义优化器再次编译它。

Answer 2

您必须使用优化器类的名称作为 custom_objects 字典中的键，在您的情况下，优化器将是“MyOptimizer”对象，

models.load_model('myModel.h5', custom_objects={'MyOptimizer': MyOptimizer})

应该管用