首页 > 解决方案 > model.load_weights() 不会恢复优化器的内部状态

问题描述

我想在构建模型后恢复优化器的状态。(不使用load_model()

我知道load_model()可以处理优化器的状态,但不幸的是,我的模型使用了很多自定义层。因此,我想通过代码构建我的模型,然后,我想恢复所有的训练状态。使用 可以恢复层中的权重model.load_weights(),但不会恢复优化器的状态。

标签: keras

解决方案


推荐阅读